From 3e433cd02d7155881dd0a2f7213a45a437eae02c Mon Sep 17 00:00:00 2001
From: Kohaku-Blueleaf <59680068+KohakuBlueleaf@users.noreply.github.com>
Date: Sat, 28 Feb 2026 00:41:20 +0800
Subject: [PATCH] Avoid inference/train tensor issue

---
 comfy/utils.py | 4 ++++
 1 file changed, 4 insertions(+)

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):