diff --git a/scripts/unsloth_install.py b/scripts/unsloth_install.py index 9c094a0ef..bbb360797 100644 --- a/scripts/unsloth_install.py +++ b/scripts/unsloth_install.py @@ -1,36 +1,6 @@ -# noqa +"""Print the uv commands required to install Unsloth.""" -try: - import torch -except ImportError as error: - raise ImportError("Install torch via `pip install torch`") from error -from packaging.version import Version as V +UNSLOTH_BASE = "uv pip install --system unsloth-zoo==2025.9.12" +UNSLOTH_HF = 'uv pip install --system --no-deps "unsloth[huggingface]==2025.9.9"' -v = V(torch.__version__) -cuda = str(torch.version.cuda) -try: - is_ampere = torch.cuda.get_device_capability()[0] >= 8 -except RuntimeError: - is_ampere = False -if cuda != "12.1" and cuda != "11.8" and cuda != "12.4": - raise RuntimeError(f"CUDA = {cuda} not supported!") -if v <= V("2.1.0"): - raise RuntimeError(f"Torch = {v} too old!") -elif v <= V("2.1.1"): - x = "cu{}{}-torch211" -elif v <= V("2.1.2"): - x = "cu{}{}-torch212" -elif v < V("2.3.0"): - x = "cu{}{}-torch220" -elif v < V("2.4.0"): - x = "cu{}{}-torch230" -elif v < V("2.5.0"): - x = "cu{}{}-torch240" -elif v < V("2.6.0"): - x = "cu{}{}-torch250" -else: - raise RuntimeError(f"Torch = {v} too new!") -x = x.format(cuda.replace(".", ""), "-ampere" if is_ampere else "") -print( - f'uv pip install --system unsloth-zoo==2024.12.1 && uv pip install --system --no-deps "unsloth[{x}]==2024.12.4"' -) +print(f"{UNSLOTH_BASE} && {UNSLOTH_HF}")