wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-01-10 14:20:49 +08:00
Code Issues Actions 11 Packages Projects Releases Wiki Activity

Update utils.py

Browse Source
This commit is contained in:
Aseem Saxena 2025-04-24 13:10:37 -07:00 committed by GitHub
parent 5fe4119d53
commit 88644341ca
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

8
comfy/utils.py
View File

@ -118,15 +118,15 @@ def state_dict_key_replace(state_dict, keys_to_replace):
return state_dict return state_dict
def state_dict_prefix_replace(state_dict, replace_prefix, filter_keys=False): def state_dict_prefix_replace(state_dict, replace_prefix, filter_keys=False):
if not filter_keys: if filter_keys:
return {} out = {}
else:
out = state_dict
for old_prefix, new_prefix in replace_prefix.items(): for old_prefix, new_prefix in replace_prefix.items():
keys_to_replace = [key for key in state_dict if key.startswith(old_prefix)] keys_to_replace = [key for key in state_dict if key.startswith(old_prefix)]
for key in keys_to_replace: for key in keys_to_replace:
new_key = new_prefix + key[len(old_prefix):] new_key = new_prefix + key[len(old_prefix):]
out[new_key] = state_dict.pop(key) out[new_key] = state_dict.pop(key)
else:
out = state_dict
return out return out