Update app.py

app.py

@@ -228,6 +228,7 @@ if torch.cuda.is_available():
     vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
     pipe = AutoPipelineForText2Image.from_pretrained(
         "Menyu/Vpred-checkpoints5000",
+        vae=vae,
         torch_dtype=torch.float16,
         use_safetensors=True,
         add_watermarker=False