Update app.py

app.py

@@ -15,11 +15,11 @@ MAX_IMG_SIZE = 4096
 
 device = torch.device("cpu")
 
-vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
+vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float32)
 pipe = DiffusionPipeline.from_pretrained(
     "stabilityai/stable-diffusion-xl-base-1.0",
     vae=vae,
-    torch_dtype=torch.float16,
+    torch_dtype=torch.float32,
     use_safetensors=True,
     variant="fp16",
 )