diff --git a/.github/workflows/main.yml b/.github/workflows/main.yml index 891300246..3daf39e43 100644 --- a/.github/workflows/main.yml +++ b/.github/workflows/main.yml @@ -98,6 +98,12 @@ jobs: 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 @@ -151,6 +157,18 @@ jobs: python_version: "3.11" pytorch: 2.6.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 + is_latest: true runs-on: axolotl-gpu-runner steps: - name: Checkout