Update app.py

app.py

@@ -8,11 +8,10 @@ import torch
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
 model_repo_id = "stabilityai/stable-diffusion-3.5-large"
+torch_dtype = torch.float16 if torch.cuda.is_available() else torch.float32
 
 if torch.cuda.is_available():
-    torch_dtype = torch.bfloat16
-else:
-    torch_dtype = torch.float32
+    torch_dtype = torch.float16 if torch.cuda.is_available() else torch.float32
 
 pipe = DiffusionPipeline.from_pretrained(model_repo_id, torch_dtype=torch_dtype)
 pipe = pipe.to(device)