Hugging Face's logo

Spaces:
prithivMLmods
/
Qwen-Image-Diffusion
Running on Zero

App Files Community
3
prithivMLmods commited on
Commit
8d08409
·
verified ·
1 Parent(s): d899f6c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -30,7 +30,8 @@ MAX_IMAGE_SIZE = 2048
30
 
31
  # Load Qwen/Qwen-Image pipeline
32
  dtype = torch.bfloat16
33
- device = "cuda" if torch.cuda.is_available() else "cpu"
 
34
  # ---- CUDA Check ----
35
  print("CUDA_VISIBLE_DEVICES=", os.environ.get("CUDA_VISIBLE_DEVICES"))
36
  print("torch.__version__ =", torch.__version__)
 
30
 
31
  # Load Qwen/Qwen-Image pipeline
32
  dtype = torch.bfloat16
33
+ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
34
+
35
  # ---- CUDA Check ----
36
  print("CUDA_VISIBLE_DEVICES=", os.environ.get("CUDA_VISIBLE_DEVICES"))
37
  print("torch.__version__ =", torch.__version__)