dtype fix

app.py

@@ -6,7 +6,6 @@ import spaces
 from diffusers import DiffusionPipeline
 import torch
 
-device = "cuda" if torch.cuda.is_available() else "cpu"
 model_repo_id = "Qwen/Qwen-Image" 
 
 MODEL_CHOICES = [
@@ -19,9 +18,11 @@ MODEL_CHOICES = [
 ]
 
 if torch.cuda.is_available():
-    torch_dtype = torch.float16
+    torch_dtype = torch.bfloat16
+    device = "cuda"
 else:
     torch_dtype = torch.float32
+    device = "cpu"
 
 pipe = DiffusionPipeline.from_pretrained(model_repo_id, torch_dtype=torch_dtype)
 pipe = pipe.to(device)