Skip to content

GitLab

Commit 5379051d authored by comfyanonymous's avatar comfyanonymous
Browse files

Fix diffusers VAE loading.

parent 00da9b32
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 0 deletions
comfy/diffusers_convert.py
View file @ 5379051d
...@@ -148,6 +148,10 @@ vae_conversion_map_attn = [ ...@@ -148,6 +148,10 @@ vae_conversion_map_attn = [
("q.", "query."), ("q.", "query."),
("k.", "key."), ("k.", "key."),
("v.", "value."), ("v.", "value."),
("q.", "to_q."),
("k.", "to_k."),
("v.", "to_v."),
("proj_out.", "to_out.0."),
("proj_out.", "proj_attn."), ("proj_out.", "proj_attn."),
] ]
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment