Update app.py

app.py

@@ -200,7 +200,6 @@ def load_and_prepare_model():
         add_watermarker=False,
         token = HF_TOKEN,
     )
-    pipe.vae = vaeXL #.to(torch.bfloat16)
     #pipe.scheduler = sched
     #pipe.vae.do_resize = False
     pipe.vae.do_convert_rgb = True
@@ -209,6 +208,7 @@ def load_and_prepare_model():
     pipe.watermark=None
     pipe.safety_checker=None  
     pipe.to(torch.device('cuda:0'), torch.bfloat16)
+    pipe.vae = vaeXL #.to(torch.bfloat16)
     #pipe.to(torch.bfloat16)
     #pipe.to(accelerator.device)
     return pipe