Update app.py

app.py

@@ -27,10 +27,10 @@ DEFAULT_WIDTH = 1024
 
 REPO = "sd-community/sdxl-flash"
 
-vae = AutoencoderKL.from_pretrained("stabilityai/sd-vae-ft-mse").to(DEVICE)
-controlnet = ControlNetModel.from_pretrained("MakiPan/controlnet-encoded-hands-130k", torch_dtype=torch.float32)
+vae = AutoencoderKL.from_pretrained("stabilityai/sd-vae-ft-mse").half().to(DEVICE)
+controlnet = ControlNetModel.from_pretrained("MakiPan/controlnet-encoded-hands-130k", torch_dtype=torch.float16)
 
-model = StableDiffusionXLPipeline.from_pretrained(REPO, vae=vae, controlnet=controlnet, torch_dtype=torch.float32, use_safetensors=True, add_watermarker=False)
+model = StableDiffusionXLPipeline.from_pretrained(REPO, vae=vae, controlnet=controlnet, torch_dtype=torch.float16, use_safetensors=True, add_watermarker=False)
 model.load_lora_weights("ehristoforu/dalle-3-xl-v2", adapter_name="base")
 model.set_adapters(["base"], adapter_weights=[0.7])
 model.to(DEVICE)