diff --git a/comfy_extras/nodes_train.py b/comfy_extras/nodes_train.py index ea8200f18..01162a28e 100644 --- a/comfy_extras/nodes_train.py +++ b/comfy_extras/nodes_train.py @@ -597,7 +597,7 @@ class TrainLoraNode: finally: for m in mp.model.modules(): unpatch(m) - del ss, train_sampler, optimizer + del train_sampler, optimizer torch.cuda.empty_cache() for adapter in all_weight_adapters: