Fix diffusers VAE loading.

5379051d · comfyanonymous · 00da9b32 · 5379051d
Commit 5379051d authored Jul 26, 2023 by comfyanonymous
Show whitespace changes
Inline Side-by-side

Showing with 4 additions and 0 deletions

comfy/diffusers_convert.py comfy/diffusers_convert.py +4 -0

No files found.
--- a/comfy/diffusers_convert.py
+++ b/comfy/diffusers_convert.py
@@ -148,6 +148,10 @@ vae_conversion_map_attn = [
    ("q.", "query."),
    ("k.", "key."),
    ("v.", "value."),
+    ("q.", "to_q."),
+    ("k.", "to_k."),
+    ("v.", "to_v."),
+    ("proj_out.", "to_out.0."),
    ("proj_out.", "proj_attn."),
 ]