Update app.py

app.py

@@ -229,7 +229,7 @@ def load_model():
     pipe.transformer=ll_transformer
     pipe.load_lora_weights("ford442/sdxl-vae-bf16", weight_name="LoRA/UltraReal.safetensors")
     pipe.to(device=device, dtype=torch.bfloat16)
-    pipe.vae=vae
+    pipe.vae=vae.to(device=device)
     upscaler_2 = UpscaleWithModel.from_pretrained("Kim2091/ClearRealityV1").to(device)
     return pipe, upscaler_2