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

coderabbito

Browse Source
This commit is contained in:
Dan Saunders
2025-09-10 16:21:17 -04:00
parent cd8c769e84
commit 4c81172917
2 changed files with 9 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -84,8 +84,9 @@ jobs:
uv pip show --system torch
uv pip install --system wheel
uv pip install --system --no-build-isolation -e ".[dev]"
python scripts/unsloth_install.py | sh
python scripts/cutcrossentropy_install.py | sh
set -o pipefail
python scripts/unsloth_install.py | bash
python scripts/cutcrossentropy_install.py | bash
- name: Make sure PyTorch version wasn't clobbered
run: |
@@ -154,10 +155,12 @@ jobs:
run: |
uv pip show --system torch
uv pip install --system wheel
uv pip install --system --no-build-isolation -e .
uv pip install --system build
python -m build --sdist
uv pip install --system dist/*.tar.gz
python scripts/unsloth_install.py | sh
python scripts/cutcrossentropy_install.py | sh
uv pip install --system -e .[dev]
uv pip install --system ".[dev]"
- name: Make sure PyTorch version wasn't clobbered
run: |

2
cicd/Dockerfile.jinja
View File

@@ -1,5 +1,7 @@
FROM axolotlai/axolotl-base-uv:{{ BASE_TAG }}
SHELL ["/bin/bash", "-euxo", "pipefail", "-c"]
ENV TORCH_CUDA_ARCH_LIST="7.0 7.5 8.0 8.6 9.0+PTX"
ENV AXOLOTL_EXTRAS="{{ AXOLOTL_EXTRAS }}"
ENV AXOLOTL_ARGS="{{ AXOLOTL_ARGS }}"