Update app.py

app.py

@@ -10,11 +10,10 @@ from diffusers import DiffusionPipeline
 device = "cuda" if torch.cuda.is_available() else "cpu"
 model_id_default = "stable-diffusion-v1-5/stable-diffusion-v1-5"  # Replace to the model you would like to use
 
-# if torch.cuda.is_available():
-#     torch_dtype = torch.float16
-# else:
-#     torch_dtype = torch.float32
-torch_dtype = torch.float32
+if torch.cuda.is_available():
+    torch_dtype = torch.float16
+else:
+    torch_dtype = torch.float32
 
 MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 1024