ovis-image-7b-cpu

Running

smartdigitalnetworks commited on 5 days ago

Commit

bc7bbfb

verified ·

1 Parent(s): f1f1ab7

Update app_diffusers.py

Files changed (1) hide show

app_diffusers.py CHANGED Viewed

@@ -16,7 +16,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")