Update app.py

app.py

@@ -121,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)
+    pipe.to(device=device, torch.bfloat16)
     pipe.vae = vaeXL.to(device) #.to('cpu') #.to(torch.bfloat16)
     
     pipe.unet.set_attn_processor(AttnProcessor2_0())