From 9d4a9ad81963d53e708f7b35bde2d393d62849f5 Mon Sep 17 00:00:00 2001
From: Kohaku-Blueleaf <59680068+KohakuBlueleaf@users.noreply.github.com>
Date: Tue, 3 Feb 2026 22:43:11 +0800
Subject: [PATCH] fix wrong input type

---
 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 18455c882..28431b030 100644
--- a/comfy_extras/nodes_train.py
+++ b/comfy_extras/nodes_train.py
@@ -1031,7 +1031,7 @@ class TrainLoraNode(io.ComfyNode):
                     max=5,
                     tooltip="Depth level for gradient checkpointing.",
                 ),
-                io.Int.Input(
+                io.Boolean.Input(
                     "offloading",
                     default=False,
                     tooltip="Depth level for gradient checkpointing.",