ovis-image-7b-cpu

Running

smartdigitalnetworks commited on 4 days ago

Commit

f1f1ab7

verified ·

1 Parent(s): 1b5c2e5

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -22,7 +22,7 @@ logging.set_verbosity_error()
 # DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
 MAX_SEED = np.iinfo(np.int32).max
-device = "cuda"
 _dtype = torch.bfloat16
 hf_token = os.getenv("HF_TOKEN")

 # DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
 MAX_SEED = np.iinfo(np.int32).max
+device = "cuda" if torch.cuda.is_available() else "cpu"
 _dtype = torch.bfloat16
 hf_token = os.getenv("HF_TOKEN")