xinyun/ComfyUI-KJNodes
1
0
Fork 0
mirror of https://git.datalinker.icu/kijai/ComfyUI-KJNodes.git synced 2025-12-09 12:54:40 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix dit lora block loader on loras with diff keys

Browse Source
This commit is contained in:
kijai 2025-08-06 23:15:12 +03:00
parent dafbcae4e6
commit 87d0cf42db
1 changed files with 16 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
nodes/nodes.py
View File

@ -1935,22 +1935,29 @@ class DiTBlockLoraLoader:
if ratio == 0:
keys_to_delete.append(key)
else:
value = loaded[key].weights
weights_list = list(loaded[key].weights)
weights_list[2] = ratio
loaded[key].weights = tuple(weights_list)
# Only modify LoRA adapters, skip diff tuples
value = loaded[key]
if hasattr(value, 'weights'):
print(f"Modifying LoRA adapter for key: {key}")
weights_list = list(value.weights)
weights_list[2] = ratio
loaded[key].weights = tuple(weights_list)
else:
print(f"Skipping non-LoRA entry for key: {key}")
for key in keys_to_delete:
del loaded[key]
print("loading lora keys:")
for key, value in loaded.items():
print(f"Key: {key}, Alpha: {value.weights[2]}")
if hasattr(value, 'weights'):
print(f"Key: {key}, Alpha: {value.weights[2]}")
else:
print(f"Key: {key}, Type: {type(value)}")
if model is not None:
new_modelpatcher = model.clone()
k = new_modelpatcher.add_patches(loaded, strength_model)
if model is not None:
new_modelpatcher = model.clone()
k = new_modelpatcher.add_patches(loaded, strength_model)
k = set(k)
for x in loaded: