fix: Default gradient checkpointing with unsloth 2026.3.3

vivekkalyan · vivekkalyan · commit 95f87d2c6df8 · 2026-03-04T18:18:22.000-08:00
diff --git a/pyproject.toml b/pyproject.toml
@@ -22,8 +22,8 @@ backend = [
     "peft>=0.14.0",
     "hf-xet>=1.1.0",
     "bitsandbytes>=0.45.2",
-    "unsloth==2025.12.9",
-    "unsloth-zoo==2025.12.7",
+    "unsloth==2026.3.3",
+    "unsloth-zoo==2026.3.1",
     "torch>=2.8.0",
     "torchao==0.15.0",
     "accelerate==1.7.0",
diff --git a/src/art/dev/get_model_config.py b/src/art/dev/get_model_config.py
@@ -60,6 +60,7 @@ def get_model_config(
         adam_beta2=0.99,
         disable_tqdm=True,
         gradient_accumulation_steps=1,
+        gradient_checkpointing=True,
         learning_rate=5e-6,
         logging_steps=1,
         lr_scheduler_type="constant",
diff --git a/uv.lock b/uv.lock
