wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-01-26 22:30:19 +08:00
Code Issues Actions 9 Packages Projects Releases Wiki Activity

Fix possibly-used-before-assignment in lora.py

Browse Source
This commit is contained in:
Max Tretikov 2024-06-13 23:46:00 -06:00
parent 5e89252ab6
commit a02c632aa8
1 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
comfy/lora.py
View File

@ -30,8 +30,9 @@ def load_lora(lora, to_load):
regular_lora = "{}.lora_up.weight".format(x) regular_lora = "{}.lora_up.weight".format(x)
diffusers_lora = "{}_lora.up.weight".format(x) diffusers_lora = "{}_lora.up.weight".format(x)
transformers_lora = "{}.lora_linear_layer.up.weight".format(x) transformers_lora = "{}.lora_linear_layer.up.weight".format(x)
A_name = None A_name = B_name = None
mid_name = None
if regular_lora in lora.keys(): if regular_lora in lora.keys():
A_name = regular_lora A_name = regular_lora
B_name = "{}.lora_down.weight".format(x) B_name = "{}.lora_down.weight".format(x)
@ -39,11 +40,9 @@ def load_lora(lora, to_load):
elif diffusers_lora in lora.keys(): elif diffusers_lora in lora.keys():
A_name = diffusers_lora A_name = diffusers_lora
B_name = "{}_lora.down.weight".format(x) B_name = "{}_lora.down.weight".format(x)
mid_name = None
elif transformers_lora in lora.keys(): elif transformers_lora in lora.keys():
A_name = transformers_lora A_name = transformers_lora
B_name ="{}.lora_linear_layer.down.weight".format(x) B_name ="{}.lora_linear_layer.down.weight".format(x)
mid_name = None
if A_name is not None: if A_name is not None:
mid = None mid = None