Update app.py

Load VAE in float16 mode

app.py

@@ -46,11 +46,16 @@ download(
     "https://civitai.com/api/download/models/290640?type=VAE&format=SafeTensor",
 )
 
-vae = AutoencoderKL.from_single_file(vae_path)
+vae = AutoencoderKL.from_single_file(
+    vae_path,
+    torch_dtype=torch.float16,
+)
 
 # pipe = StableDiffusionXLImg2ImgPipeline.from_single_file(
 pipe = StableDiffusionXLPipeline.from_single_file(
-    model_path, torch_dtype=torch.float16, use_safetensors=True, variant="fp16", vae=vae
+    model_path,
+    torch_dtype=torch.float16,
+    vae=vae,
 )
 pipe = pipe.to("cuda")