Upload app.py with huggingface_hub

app.py

@@ -1,6 +1,6 @@
 import gradio as gr
 import spaces
-from diffusers import StableDiffusionXLPipeline
+from diffusers import StableDiffusionXLPipeline, AutoencoderKL
 import numpy as np
 import math
 import torch
@@ -12,9 +12,11 @@ theme = gr.themes.Base(
     font=[gr.themes.GoogleFont('Libre Franklin'), gr.themes.GoogleFont('Public Sans'), 'system-ui', 'sans-serif'],
 )
 
+vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
 pipe = StableDiffusionXLPipeline.from_pretrained(
     "stabilityai/stable-diffusion-xl-base-1.0",
     custom_pipeline="multimodalart/sdxl_perturbed_attention_guidance",
+    vae=vae,
     torch_dtype=torch.float16,
     trust_remote_code=True
 )