298977887/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-04-20 03:13:30 +00:00
Code Issues Packages Projects Releases Wiki Activity

Small fix for SD2.x loras.

Browse Source
This commit is contained in:
comfyanonymous 2023-02-05 11:38:25 -05:00
parent 678105fade
commit bf9ccffb17
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
comfy/sd.py
View File

@ -153,8 +153,8 @@ def model_lora_keys(model, key_map={}):
key_map[lora_key] = (k, 0) key_map[lora_key] = (k, 0)
k = "model.transformer.resblocks.{}.attn.in_proj_weight".format(b) k = "model.transformer.resblocks.{}.attn.in_proj_weight".format(b)
if k in sdk: if k in sdk:
key_map[text_model_lora_key.format(b, "self_attn_k_proj")] = (k, 0) key_map[text_model_lora_key.format(b, "self_attn_q_proj")] = (k, 0)
key_map[text_model_lora_key.format(b, "self_attn_q_proj")] = (k, 1) key_map[text_model_lora_key.format(b, "self_attn_k_proj")] = (k, 1)
key_map[text_model_lora_key.format(b, "self_attn_v_proj")] = (k, 2) key_map[text_model_lora_key.format(b, "self_attn_v_proj")] = (k, 2)
return key_map return key_map