wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-03-17 23:25:05 +08:00
Code Issues Actions 18 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 eb33188c8e
commit 3e433cd02d
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
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))
def copy_to_param(obj, attr, value):