alphard/efficiency-nodes-comfyui
1
0
Fork 0
mirror of https://github.com/jags111/efficiency-nodes-comfyui.git synced 2026-03-23 14:12:11 -03:00
Code Issues Packages Projects Releases Wiki Activity

Lora Loader Fix + Efficiency Loader Clip re-add

Browse Source
This commit is contained in:
TSC
2023-07-08 10:29:21 -05:00
committed by GitHub
parent de9a7d4f08
commit 45dd318140
2 changed files with 8 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
lora_patch.py
View File

@@ -46,7 +46,7 @@ LORA_UNET_MAP_RESNET = {
"skip_connection": "resnets_{}_conv_shortcut"
}
def load_lora(path, to_load):
def load_lora_legacy(path, to_load):
lora = utils.load_torch_file(path)
patch_dict = {}
loaded_keys = set()
@@ -244,10 +244,10 @@ def model_lora_keys(model, key_map={}):
return key_map
def load_lora_for_models(model, clip, lora_path, strength_model, strength_clip):
def load_lora_for_models_legacy(model, clip, lora_path, strength_model, strength_clip):
key_map = model_lora_keys(model.model)
key_map = model_lora_keys(clip.cond_stage_model, key_map)
loaded = load_lora(lora_path, key_map)
loaded = load_lora_legacy(lora_path, key_map)
new_modelpatcher = model.clone()
k = new_modelpatcher.add_patches(loaded, strength_model)
new_clip = clip.clone()