diff --git a/comfy/utils.py b/comfy/utils.py index 0769cef44..f77acbdda 100644 --- a/comfy/utils.py +++ b/comfy/utils.py @@ -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):