fix wrong input type

2026-02-06 19:42:34 +08:00 · 2026-02-03 22:43:11 +08:00 · 2026-02-03 22:43:11 +08:00 · 9d4a9ad819
commit 9d4a9ad819
parent 20bd2c0236
1 changed files with 1 additions and 1 deletions
--- 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.",