constrain torch version

This commit is contained in:
Dan Saunders
2025-09-26 13:20:45 -04:00
parent b4c6675cd2
commit 2e082d47cc
2 changed files with 25 additions and 6 deletions

View File

@@ -1,6 +1,23 @@
"""Print the uv commands required to install Unsloth."""
"""Print the uv commands required to install Unsloth without altering Torch."""
UNSLOTH_BASE = "uv pip install --system unsloth-zoo==2025.9.12"
UNSLOTH_HF = 'uv pip install --system --no-deps "unsloth[huggingface]==2025.9.9"'
try:
import torch
except ImportError as error:
raise ImportError("Install torch via `pip install torch`") from error
print(f"{UNSLOTH_BASE} && {UNSLOTH_HF}")
from packaging.version import Version as V
TORCH_MIN = V("2.6.0")
UNSLOTH_BASE = (
"uv pip install --system --no-deps unsloth-zoo==2025.9.12"
' && uv pip install --system --no-deps "unsloth[huggingface]==2025.9.9"'
)
version = V(torch.__version__)
if version < TORCH_MIN:
raise RuntimeError(
f"Torch {version} detected, but Unsloth requires >= {TORCH_MIN}. "
"Upgrade your torch install and re-run this helper."
)
print(UNSLOTH_BASE)