diff --git a/.github/workflows/main.yml b/.github/workflows/main.yml index f34a0cf2f..052f9aa72 100644 --- a/.github/workflows/main.yml +++ b/.github/workflows/main.yml @@ -15,21 +15,6 @@ jobs: fail-fast: false matrix: include: - - cuda: 126 - cuda_version: 12.6.3 - python_version: "3.11" - pytorch: 2.7.0 - axolotl_extras: - - cuda: 126 - cuda_version: 12.6.3 - python_version: "3.11" - pytorch: 2.7.1 - axolotl_extras: vllm - - cuda: 128 - cuda_version: 12.8.1 - python_version: "3.11" - pytorch: 2.7.1 - axolotl_extras: - cuda: 128 cuda_version: 12.8.1 python_version: "3.11" @@ -46,6 +31,11 @@ jobs: python_version: "3.11" pytorch: 2.9.1 axolotl_extras: + - cuda: 130 + cuda_version: 13.0.0 + python_version: "3.11" + pytorch: 2.9.1 + axolotl_extras: runs-on: axolotl-gpu-runner steps: - name: Checkout @@ -92,27 +82,6 @@ jobs: strategy: matrix: include: - - cuda: 126 - cuda_version: 12.6.3 - python_version: "3.11" - pytorch: 2.7.0 - axolotl_extras: - - cuda: 126 - cuda_version: 12.6.3 - python_version: "3.11" - pytorch: 2.7.1 - axolotl_extras: - is_latest: - - cuda: 126 - cuda_version: 12.6.3 - python_version: "3.11" - pytorch: 2.7.1 - axolotl_extras: vllm - - cuda: 128 - cuda_version: 12.8.1 - python_version: "3.11" - pytorch: 2.7.1 - axolotl_extras: - cuda: 128 cuda_version: 12.8.1 python_version: "3.11" @@ -129,6 +98,11 @@ jobs: python_version: "3.11" pytorch: 2.9.1 axolotl_extras: + - cuda: 130 + cuda_version: 13.0.0 + python_version: "3.11" + pytorch: 2.9.1 + axolotl_extras: runs-on: axolotl-gpu-runner steps: - name: Checkout @@ -170,24 +144,18 @@ jobs: strategy: matrix: include: - - cuda: 126 - cuda_version: 12.6.3 - python_version: "3.11" - pytorch: 2.7.1 - axolotl_extras: - is_latest: - - cuda: 126 - cuda_version: 12.6.3 - python_version: "3.11" - pytorch: 2.7.1 - axolotl_extras: vllm - is_latest: true - cuda: 128 cuda_version: 12.8.1 python_version: "3.11" pytorch: 2.8.0 axolotl_extras: is_latest: + - cuda: 128 + cuda_version: 12.8.1 + python_version: "3.11" + pytorch: 2.9.1 + axolotl_extras: + is_latest: runs-on: axolotl-gpu-runner steps: - name: Checkout diff --git a/.nojekyll b/.nojekyll index 8339d6d94..ae66ff82b 100644 --- a/.nojekyll +++ b/.nojekyll @@ -1 +1 @@ -4e8b247b \ No newline at end of file +984e3917 \ No newline at end of file diff --git a/docs/docker.html b/docs/docker.html index fb553661b..20af18ef6 100644 --- a/docs/docker.html +++ b/docs/docker.html @@ -816,11 +816,8 @@ Important
main-base-py{python_version}-cu{cuda_version}-{pytorch_version}Tags examples:
main-base-py3.11-cu128-2.7.1main-base-py3.11-cu126-2.7.1main-base-py3.11-cu126-2.7.0main-base-py3.11-cu126-2.6.0main-base-py3.11-cu124-2.6.0main-base-py3.11-cu128-2.8.0main-base-py3.11-cu128-2.9.1Tags examples:
main-py3.11-cu128-2.7.1main-py3.11-cu126-2.7.1main-py3.11-cu126-2.7.0main-py3.11-cu126-2.6.0main-py3.11-cu124-2.6.0main-py3.11-cu128-2.8.0main-py3.11-cu128-2.9.1main-latestmain-20250303-py3.11-cu124-2.6.0main-20250303-py3.11-cu126-2.6.00.10.10.12.0For Blackwell GPUs, please use Pytorch 2.7.0 and CUDA 12.8.
+For Blackwell GPUs, please use Pytorch 2.9.1 and CUDA 12.8.
For Blackwell GPUs, please use axolotlai/axolotl:main-py3.11-cu128-2.7.0 or the cloud variant axolotlai/axolotl-cloud:main-py3.11-cu128-2.7.0.
For Blackwell GPUs, please use axolotlai/axolotl:main-py3.11-cu128-2.9.1 or the cloud variant axolotlai/axolotl-cloud:main-py3.11-cu128-2.9.1.
Please refer to the Docker documentation for more information on the different Docker images that are available.
diff --git a/index.html b/index.html index d507c5532..fd31ed7cb 100644 --- a/index.html +++ b/index.html @@ -844,7 +844,7 @@ Expand older updatesbf16 and Flash Attention) or AMD GPU