298977887/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-01-25 15:55:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix diffusers VAE loading.

Browse Source
This commit is contained in:
comfyanonymous 2023-07-26 18:26:39 -04:00
parent 00da9b3268
commit 5379051d16
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
comfy/diffusers_convert.py
View File

@ -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."),
] ]