wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-01-25 05:40:15 +08:00
Code Issues Actions 15 Packages Projects Releases Wiki Activity

make setattr safe for non existent attributes

Browse Source 
Handle the case where the attribute doesnt exist by returning a static
sentinel (distinct from None). If the sentinel is passed in as the set
value, del the attr.
This commit is contained in:
Rattus 2025-12-07 20:53:01 +10:00
parent 3b0368aa34
commit 482a9d58a8
1 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
comfy/utils.py
View File

@ -803,12 +803,17 @@ def safetensors_header(safetensors_path, max_size=100*1024*1024):
return None return None
return f.read(length_of_header) return f.read(length_of_header)
ATTR_UNSET={}
def set_attr(obj, attr, value): def set_attr(obj, attr, value):
attrs = attr.split(".") attrs = attr.split(".")
for name in attrs[:-1]: for name in attrs[:-1]:
obj = getattr(obj, name) obj = getattr(obj, name)
prev = getattr(obj, attrs[-1]) prev = getattr(obj, attrs[-1], ATTR_UNSET)
setattr(obj, attrs[-1], value) if value is ATTR_UNSET:
delattr(obj, attrs[-1])
else:
setattr(obj, attrs[-1], value)
return prev return prev
def set_attr_param(obj, attr, value): def set_attr_param(obj, attr, value):