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.",