Update app.py

app.py

@@ -106,7 +106,6 @@ def load_and_prepare_model():
     #pipe.vae.vae_scale_factor=8
     #pipe.to(device)
     #pipe.to(torch.bfloat16)
-    pipe.vae.set_default_attn_processor()
     print(f'init noise scale: {pipe.scheduler.init_noise_sigma}')
     pipe.watermark=None
     pipe.safety_checker=None
@@ -114,8 +113,8 @@ def load_and_prepare_model():
     #pipe.enable_vae_tiling()
     pipe.to(device=device, dtype=torch.bfloat16)
     pipe.unet.set_attn_processor(AttnProcessor2_0())
-    
     pipe.vae = vaeXL.to('cpu') #.to(torch.bfloat16)
+    pipe.vae.set_default_attn_processor()
     return pipe
     
 pipe = load_and_prepare_model()