Update app.py

app.py

@@ -118,7 +118,7 @@ def load_and_prepare_model():
     pipe.watermark=None
     pipe.safety_checker=None
     #pipe.unet.to(memory_format=torch.channels_last)
-    pipe.to(device=device)
+    pipe.to(device,torch.bfloat16)
     pipe.vae = vaeXL.to(device) #.to('cpu') #.to(torch.bfloat16)
     pipe.vae.set_default_attn_processor()
     return pipe
@@ -136,7 +136,7 @@ refiner = StableDiffusionXLImg2ImgPipeline.from_pretrained(
 refiner.vae=vaeX
 refiner.vae.set_default_attn_processor()
 
-refiner.to(device=device)
+refiner.to(device,torch.bfloat16)
 
 MAX_SEED = np.iinfo(np.int32).max