Update app.py

app.py

@@ -11,9 +11,10 @@ Path("tmp").mkdir(exist_ok=True)
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
 print(f"Device is {device}")
+torch_type = torch.float16 if device == "cuda" else torch.float32
 pipe = StableDiffusionLDM3DPipeline.from_pretrained(
     "Intel/ldm3d-4c",
-    torch_dtype=torch.float16
+    torch_dtype=torch_type
     # , safety_checker=None
 )
 pipe.to(device)