Fix AsianRealistic rainbow noise: override VAE with sd-vae-ft-mse (#20)

- Fix AsianRealistic rainbow noise: override VAE with sd-vae-ft-mse (66482eb213c52c965c34ad2ea91a9d429b4ea7d3)


Co-authored-by: pormungtailaw <pormungtai@users.noreply.huggingface.co>

models.json

@@ -66,6 +66,7 @@
       "type": "checkpoint",
       "repo_id": "stablediffusionapi/asianrealisticsdlifechias",
       "single_file_url": null,
+      "vae": "stabilityai/sd-vae-ft-mse",
       "preview": "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/9c67da8b-5e57-413c-a5ca-135f5ed9af18/width=450/1743066.jpeg",
       "trigger": "",
       "clip_skip": 2,

pipeline_manager.py

@@ -262,6 +262,12 @@ def _build_base_pipeline(cfg):
         except Exception as e:  # noqa
             print(f"[lora] fuse skipped: {e}")
 
+    # Optional VAE override — some merged checkpoints ship a broken fp16 VAE that
+    # decodes to rainbow noise; a known-good VAE fixes it.
+    if cfg.get("vae"):
+        from diffusers import AutoencoderKL
+        pipe.vae = AutoencoderKL.from_pretrained(cfg["vae"], torch_dtype=DTYPE_SD)
+
     # SD1.5 / SDXL community checkpoints are tuned for the Euler Ancestral sampler;
     # it matches the look people get in A1111 / ComfyUI far better than the default.
     if base in ("sd15", "sdxl"):