Fix VAE dtype mismatch (fp16)

app.py

@@ -209,7 +209,7 @@ def save_cropped(imgs, out_path: str):
     out = np.concatenate(cropped, axis=1)
     os.makedirs(os.path.dirname(out_path), exist_ok=True)
     imageio.imsave(out_path, out)
-    
+
 from diffusers import UniPCMultistepScheduler, AutoencoderKL, UNet2DConditionModel
 
 @lru_cache(maxsize=1)
@@ -348,6 +348,18 @@ def run_one(paths: Paths, prompt: str, steps: int = DEFAULT_STEPS):
         garment_images=garment_pil,
         garment_mask=garment_mask_pil,
     )
+    
+ 
+    if device == "cuda":
+        pipe.to(dtype=torch.float16)
+        try:
+            for _, proc in pipe.unet.attn_processors.items():
+                proc.to(dtype=torch.float16)
+        except Exception:
+            pass
+
+
+
 
     style_img = Image.open(paths.style_path).convert("RGB")