pin unsloth to a specific release

axolotl-ai-cloud · Nov 22, 2024 · a90e5cd · a90e5cd
1 parent 80f797c
commit a90e5cd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/scripts/unsloth_install.py b/scripts/unsloth_install.py
@@ -32,5 +32,5 @@
     raise RuntimeError(f"Torch = {v} too new!")
 x = x.format(cuda.replace(".", ""), "-ampere" if is_ampere else "")
 print(
-    f'pip install unsloth-zoo && pip install --no-deps "unsloth[{x}] @ git+https://github.com/unslothai/unsloth.git"'
+    f'pip install unsloth-zoo==2024.11.7 && pip install --no-deps "unsloth[{x}]==2024.11.9"'
 )