Update app.py

app.py

@@ -118,7 +118,7 @@ def load_and_prepare_model():
         token=HF_TOKEN,
         text_encoder=None,
         text_encoder_2=None,
-        vae=None,
+        vae=vaeX,
     )
     
     '''
@@ -137,7 +137,7 @@ def load_and_prepare_model():
     '''
     
     pipe.to(device=device, dtype=torch.bfloat16)
-    pipe.vae = vaeX #.to(torch.bfloat16)
+    pipe.vae.to('cpu')
     pipe.vae.do_resize=False
     #pipe.vae.do_rescale=False
     #pipe.vae.do_convert_rgb=True