RomanNum3ral
/
ComfyUI
mirror of https://github.com/Comfy-Org/ComfyUI.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Ensure conditions are not trainable to avoid bugs (#14368)

This commit is contained in:
Kohaku-Blueleaf 2026-06-10 03:55:29 +08:00 committed by GitHub
parent 6f01b244a2
commit ad564899d3
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
comfy_extras/nodes_train.py
View File

@ -15,6 +15,7 @@ import comfy.sampler_helpers
import comfy.sd import comfy.sd
import comfy.utils import comfy.utils
import comfy.model_management import comfy.model_management
from comfy.conds import CONDRegular, CONDList
from comfy.cli_args import args, PerformanceFeature from comfy.cli_args import args, PerformanceFeature
import comfy_extras.nodes_custom_sampler import comfy_extras.nodes_custom_sampler
import folder_paths import folder_paths
@ -120,6 +121,11 @@ def process_cond_list(d, prefix=""):
process_cond_list(v, f"{prefix}.{k}") process_cond_list(v, f"{prefix}.{k}")
elif isinstance(v, torch.Tensor): elif isinstance(v, torch.Tensor):
d[k] = v.clone() d[k] = v.clone()
elif isinstance(v, CONDList):
v.cond = [t.detach() if isinstance(t, torch.Tensor) else t for t in v.cond]
elif isinstance(v, CONDRegular):
if isinstance(v.cond, torch.Tensor):
v.cond = v.cond.detach()
elif isinstance(v, (list, tuple)): elif isinstance(v, (list, tuple)):
for index, item in enumerate(v): for index, item in enumerate(v):
process_cond_list(item, f"{prefix}.{k}.{index}") process_cond_list(item, f"{prefix}.{k}.{index}")