From 3e3adfe804d7b4eea122c748df78b0f36b3d9be1 Mon Sep 17 00:00:00 2001
From: Kohaku-Blueleaf <59680068+KohakuBlueleaf@users.noreply.github.com>
Date: Sat, 28 Jun 2025 23:10:39 +0800
Subject: [PATCH] Update nodes_train.py

---
 comfy_extras/nodes_train.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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: