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

Merge 88644341ca into 5ac3b26a7d

Browse Source
This commit is contained in:
Aseem Saxena 2025-12-15 10:58:01 +08:00 committed by GitHub
commit a4b8632c34
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
comfy/utils.py
View File

@ -135,13 +135,13 @@ def state_dict_key_replace(state_dict, keys_to_replace):
def state_dict_prefix_replace(state_dict, replace_prefix, filter_keys=False):
if filter_keys:
out = {}
for old_prefix, new_prefix in replace_prefix.items():
keys_to_replace = [key for key in state_dict if key.startswith(old_prefix)]
for key in keys_to_replace:
new_key = new_prefix + key[len(old_prefix):]
out[new_key] = state_dict.pop(key)
else:
out = state_dict
for rp in replace_prefix:
replace = list(map(lambda a: (a, "{}{}".format(replace_prefix[rp], a[len(rp):])), filter(lambda a: a.startswith(rp), state_dict.keys())))
for x in replace:
w = state_dict.pop(x[0])
out[x[1]] = w
return out