Various bug fixes

2026-04-29 18:51:37 +00:00 · 2025-04-29 09:30:33 -06:00
parent 2b4c525489
commit 5890e67a46
3 changed files with 8 additions and 10 deletions
--- a/toolkit/util/vae.py
+++ b/toolkit/util/vae.py
@@ -10,7 +10,7 @@ def load_vae(vae_path, dtype):
    except Exception as e:
        try:
            vae = AutoencoderKL.from_pretrained(
-                vae_path.vae_path,
+                vae_path,
                subfolder="vae",
                torch_dtype=dtype,
            )