Skip to content

Commit ca17fc8

Browse files
Fix potential issue. (Comfy-Org#13009)
1 parent 20561aa commit ca17fc8

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

comfy/utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -899,7 +899,7 @@ def set_attr(obj, attr, value):
899899
def set_attr_param(obj, attr, value):
900900
# Clone inference tensors (created under torch.inference_mode) since
901901
# their version counter is frozen and nn.Parameter() cannot wrap them.
902-
if value.is_inference():
902+
if (not torch.is_inference_mode_enabled()) and value.is_inference():
903903
value = value.clone()
904904
return set_attr(obj, attr, torch.nn.Parameter(value, requires_grad=False))
905905

0 commit comments

Comments
 (0)