Various bug fixes

2025-04-29 09:30:33 -06:00
1 parent 2b4c525489
commit 5890e67a46
@@ -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,
            )