From 553134730705e23c24e24eb38dfc5f6d950e3c35 Mon Sep 17 00:00:00 2001 From: oobabooga Date: Wed, 15 Jul 2026 10:19:20 -0300 Subject: [PATCH] Studio: honor the 'none' gradient checkpointing option in training (#7128) --- studio/backend/core/training/trainer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/studio/backend/core/training/trainer.py b/studio/backend/core/training/trainer.py index 96d1b90b16..932bcaf0d5 100644 --- a/studio/backend/core/training/trainer.py +++ b/studio/backend/core/training/trainer.py @@ -931,7 +931,7 @@ class UnslothTrainer: use_gradient_checkpointing = "unsloth" elif use_gradient_checkpointing in ("true", "1", "yes"): use_gradient_checkpointing = True - elif use_gradient_checkpointing in ("false", "0", "no"): + elif use_gradient_checkpointing in ("false", "0", "no", "none", "off"): use_gradient_checkpointing = False else: # Invalid value -> "unsloth"