From acc881452f11b7ec686c8a24d356fe014ef97551 Mon Sep 17 00:00:00 2001
From: Daniel Han <danielhanchen@gmail.com>
Date: Tue, 24 Mar 2026 07:44:07 -0700
Subject: [PATCH] fix: pin unsloth>=2026.3.11 in install.sh and install.ps1
 (#4556)

Ensures both install scripts always pull a version that has the
litellm removal fix. Without the pin, stale uv/pip caches could
resolve the older 2026.3.10 which still had litellm in
data-designer-deps.txt, causing setup to fail at step 8/11
while PyPI has litellm quarantined.
---
 install.ps1 | 2 +-
 install.sh  | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/install.ps1 b/install.ps1
index d438e5ed2d..52386dd2fd 100644
--- a/install.ps1
+++ b/install.ps1
@@ -275,7 +275,7 @@ function Install-UnslothStudio {
     }
 
     Write-Host "==> Installing unsloth (this may take a few minutes)..."
-    uv pip install --python $VenvPython --upgrade-package unsloth unsloth
+    uv pip install --python $VenvPython --upgrade-package unsloth "unsloth>=2026.3.11"
     if ($LASTEXITCODE -ne 0) {
         Write-Host "[ERROR] Failed to install unsloth (exit code $LASTEXITCODE)" -ForegroundColor Red
         return
diff --git a/install.sh b/install.sh
index 18db3469f4..3d8e08612c 100755
--- a/install.sh
+++ b/install.sh
@@ -235,7 +235,7 @@ fi
 
 # ── Install unsloth directly into the venv (no activation needed) ──
 echo "==> Installing unsloth (this may take a few minutes)..."
-uv pip install --python "$VENV_NAME/bin/python" unsloth --torch-backend=auto
+uv pip install --python "$VENV_NAME/bin/python" "unsloth>=2026.3.11" --torch-backend=auto
 
 # ── Run studio setup ──
 # Ensure the venv's Python is on PATH for setup.sh's Python discovery.