Added vape to reduce vram usage

app.py

@@ -1,13 +1,15 @@
-from diffusers import StableDiffusionXLPipeline
+from diffusers import StableDiffusionXLPipeline, AutoencoderKL
 import torch
 #from controlnet_aux import OpenposeDetector
 #from diffusers.utils import load_image
 import gradio as gr
 
+
 model_base = "stabilityai/stable-diffusion-xl-base-1.0"
 
+vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
 pipe = StableDiffusionXLPipeline.from_pretrained(
-    model_base, torch_dtype=torch.float16
+    model_base, vae=vae, torch_dtype=torch.float16, variant="fp16", use_safetensors=True
 )
 pipe = pipe.to("cuda")