Update app.py

app.py

@@ -100,15 +100,16 @@ def load_and_prepare_model():
         add_watermarker=False,
         #text_encoder=None,
         #text_encoder_2=None,
-        unet=UNet2DConditionModel.from_pretrained(
+        unet=None,
+        vae=None,
+    )
+    pipe.unet=UNet2DConditionModel.from_pretrained(
             'ford442/RealVisXL_V5.0_BF16',
             low_cpu_mem_usage=False,
             subfolder='unet',
             upcast_attention=True,
             #attention_type='gated-text-image',
-            token=True),
-        vae=None,
-    )
+            token=True)
     #pipe.vae = vaeXL #.to(torch.bfloat16)
     #pipe.scheduler = sched
     #pipe.vae.do_resize=False
@@ -120,7 +121,7 @@ def load_and_prepare_model():
     pipe.safety_checker=None
     #pipe.unet.to(memory_format=torch.channels_last)
     #pipe.enable_vae_tiling()
-    pipe.to(device=device, dtype=torch.bfloat16)
+    pipe.to(device=device)
     pipe.vae = vaeXL.to(device) #.to('cpu') #.to(torch.bfloat16)
     
     pipe.unet.set_attn_processor(AttnProcessor2_0())