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

constrain torch version

Browse Source
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
Download Patch File Download Diff File Expand all files Collapse all files

6
.github/workflows/tests.yml vendored
View File

@@ -83,7 +83,8 @@ jobs:
run: |
uv pip show --system torch
uv pip install --system wheel
uv pip install --system --no-build-isolation -e ".[dev]"
printf "torch==${{ matrix.pytorch_version }}\n" > torch-constraints.txt
uv pip install --system --no-build-isolation -e ".[dev]" --constraints torch-constraints.txt
set -o pipefail
python scripts/unsloth_install.py | bash
python scripts/cutcrossentropy_install.py | bash
@@ -160,7 +161,8 @@ jobs:
uv pip install --system dist/*.tar.gz
python scripts/unsloth_install.py | sh
python scripts/cutcrossentropy_install.py | sh
uv pip install --system ".[dev]"
printf "torch==${{ matrix.pytorch_version }}\n" > torch-constraints.txt
uv pip install --system ".[dev]" --constraints torch-constraints.txt
- name: Make sure PyTorch version wasn't clobbered
run: |