Update app_diffusers.py

app_diffusers.py

@@ -13,10 +13,10 @@ from diffusers import OvisImagePipeline
 
 logging.set_verbosity_error()
 
-DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
+# 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"
+device = "cuda" if torch.cuda.is_available() else "cpu"
 _dtype = torch.bfloat16
 hf_token = os.getenv("HF_TOKEN")