Update app.py

app.py

@@ -18,6 +18,9 @@ from PIL import Image
 
 base_model="stabilityai/stable-diffusion-xl-base-1.0"
 
+# base = DiffusionPipeline.from_pretrained(base_model,
+#         torch_dtype=torch.float16, use_safetensors=True
+# ).to("cuda")
 base = DiffusionPipeline.from_pretrained(base_model,
         torch_dtype=torch.float16, use_safetensors=True
 ).to("cuda")