Update app.py

app.py

@@ -13,6 +13,7 @@ unet = UNet2DConditionModel.from_config("SG161222/Realistic_Vision_V5.0_noVAE",
 unet.load_state_dict(load_file(hf_hub_download("ByteDance/SDXL-Lightning", "sdxl_lightning_2step_unet.safetensors"), device="cuda"))
 vae = AutoencoderKL.from_pretrained("stabilityai/sd-vae-ft-mse")
 pipe = AutoPipelineForInpainting.from_pretrained("SG161222/Realistic_Vision_V5.0_noVAE",unet=unet,vae=vae).to("cuda")
+unet = unet.to(dtype=torch.float16)
 
 def read_content(file_path: str) -> str:
     """read the content of target file