build: Upgrade to unsloth 2026.3.3 (#604)

vivekkalyan · web-flow · commit 43a6ed084ca9 · 2026-03-04T18:52:42.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
