fix vae

app_masked.py

@@ -10,8 +10,11 @@ from segment_utils import(
 )
 from diffusers import (
     DiffusionPipeline,
+    AutoencoderKL,
 )
 
+vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
+
 BASE_MODEL = "RunDiffusion/Juggernaut-XL-v9"
 
 DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
@@ -23,8 +26,9 @@ DEFAULT_CATEGORY = "hair"
 
 basepipeline = DiffusionPipeline.from_pretrained(
     BASE_MODEL,
+    vae=vae,
     torch_dtype=torch.float16,
-    # use_safetensors=True,
+    use_safetensors=True,
     custom_pipeline="./pipelines/masked_stable_diffusion_xl_img2img.py",
 )