wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-03-23 01:53:32 +08:00
Code Issues Actions 8 Packages Projects Releases Wiki Activity

Avoid inference/train tensor issue

Browse Source
This commit is contained in:
Kohaku-Blueleaf 2026-02-28 00:41:20 +08:00
parent 6e2a2ee342
commit 3690e8134f
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
comfy/utils.py
View File

@ -881,6 +881,10 @@ def set_attr(obj, attr, value):
return prev return prev
def set_attr_param(obj, attr, value): def set_attr_param(obj, attr, value):
# Clone inference tensors (created under torch.inference_mode) since
# their version counter is frozen and nn.Parameter() cannot wrap them.
if value.is_inference():
value = value.clone()
return set_attr(obj, attr, torch.nn.Parameter(value, requires_grad=False)) return set_attr(obj, attr, torch.nn.Parameter(value, requires_grad=False))
def copy_to_param(obj, attr, value): def copy_to_param(obj, attr, value):