tocmo0nlord/axolotl
1
0
Fork 0
Code Issues Pull Requests Actions 3 Packages Projects Releases Wiki Activity

git SHA

Browse Source
This commit is contained in:
Dan Saunders
2025-09-26 19:39:08 -04:00
parent cec2490903
commit 26a58bb8af
2 changed files with 21 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
.github/workflows/main.yml vendored
View File

@@ -68,6 +68,8 @@ jobs:
PYTORCH_VERSION=${{ matrix.pytorch }}
AXOLOTL_ARGS=${{ matrix.axolotl_args }}
AXOLOTL_EXTRAS=${{ matrix.axolotl_extras}}
GIT_REF=${{ github.ref }}
GIT_SHA=${{ github.sha }}
file: ./docker/Dockerfile
push: ${{ github.event_name != 'pull_request' }}
tags: |
@@ -205,6 +207,8 @@ jobs:
build-args: |
BASE_TAG=${{ github.ref_type == 'tag' && 'main' || github.ref_name }}-py${{ matrix.python_version }}-cu${{ matrix.cuda }}-${{ matrix.pytorch }}${{ matrix.axolotl_extras != '' && '-' || '' }}${{ matrix.axolotl_extras }}
CUDA=${{ matrix.cuda }}
GIT_REF=${{ github.ref }}
GIT_SHA=${{ github.sha }}
file: ./docker/Dockerfile-cloud-no-tmux
push: ${{ github.event_name != 'pull_request' }}
tags: |

30
scripts/unsloth_install.py
View File

@@ -1,23 +1,27 @@
"""Print the uv commands required to install Unsloth without altering Torch."""
"""Emit the uv commands needed to install Unsloth without touching torch."""
from __future__ import annotations
try:
import torch
except ImportError as error:
raise ImportError("Install torch via `pip install torch`") from error
except ImportError as exc: # pragma: no cover
raise ImportError("Install torch via `pip install torch`") from exc
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"'
)
torch_version = V(torch.__version__.split("+")[0])
version = V(torch.__version__)
if version < TORCH_MIN:
# Unsloth supports torch >= 2.6.0 via the 2025.9 builds.
MIN_TORCH = V("2.6.0")
if torch_version < MIN_TORCH:
raise RuntimeError(
f"Torch {version} detected, but Unsloth requires >= {TORCH_MIN}. "
"Upgrade your torch install and re-run this helper."
f"Torch {torch.__version__} detected, but Unsloth requires >= {MIN_TORCH}."
)
print(UNSLOTH_BASE)
commands = (
"uv pip install --system --no-deps unsloth-zoo==2025.9.12 && "
'uv pip install --system --no-deps "unsloth[huggingface]==2025.9.9"'
)
print(commands)