diff --git a/.github/workflows/main.yml b/.github/workflows/main.yml index 47a4c7f114..8f96461550 100644 --- a/.github/workflows/main.yml +++ b/.github/workflows/main.yml @@ -32,7 +32,7 @@ jobs: - cuda: 124 cuda_version: 12.4.1 python_version: "3.11" - pytorch: 2.5.0 + pytorch: 2.5.1 axolotl_extras: runs-on: axolotl-gpu-runner steps: @@ -94,7 +94,7 @@ jobs: - cuda: 124 cuda_version: 12.4.1 python_version: "3.11" - pytorch: 2.5.0 + pytorch: 2.5.1 axolotl_extras: runs-on: axolotl-gpu-runner steps: diff --git a/.github/workflows/multi-gpu-e2e.yml b/.github/workflows/multi-gpu-e2e.yml index d9f0ce7e6c..c2fb5dfb5d 100644 --- a/.github/workflows/multi-gpu-e2e.yml +++ b/.github/workflows/multi-gpu-e2e.yml @@ -31,7 +31,7 @@ jobs: - cuda: 124 cuda_version: 12.4.1 python_version: "3.11" - pytorch: 2.5.0 + pytorch: 2.5.1 axolotl_extras: num_gpus: 2 nightly_build: "true" diff --git a/.github/workflows/nightlies.yml b/.github/workflows/nightlies.yml index 55123a9026..3d6c0e62d7 100644 --- a/.github/workflows/nightlies.yml +++ b/.github/workflows/nightlies.yml @@ -31,7 +31,7 @@ jobs: - cuda: 124 cuda_version: 12.4.1 python_version: "3.11" - pytorch: 2.5.0 + pytorch: 2.5.1 axolotl_extras: runs-on: axolotl-gpu-runner steps: @@ -93,7 +93,7 @@ jobs: - cuda: 124 cuda_version: 12.4.1 python_version: "3.11" - pytorch: 2.5.0 + pytorch: 2.5.1 axolotl_extras: runs-on: axolotl-gpu-runner steps: diff --git a/.github/workflows/tests-nightly.yml b/.github/workflows/tests-nightly.yml index 6a1dbcc915..cba39b94e7 100644 --- a/.github/workflows/tests-nightly.yml +++ b/.github/workflows/tests-nightly.yml @@ -25,7 +25,7 @@ jobs: fail-fast: false matrix: python_version: ["3.10", "3.11"] - pytorch_version: ["2.3.1", "2.4.1", "2.5.0"] + pytorch_version: ["2.3.1", "2.4.1", "2.5.1"] timeout-minutes: 20 steps: @@ -92,7 +92,7 @@ jobs: - cuda: 124 cuda_version: 12.4.1 python_version: "3.11" - pytorch: 2.5.0 + pytorch: 2.5.1 num_gpus: 1 axolotl_extras: nightly_build: "true" diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml index 2c735bd102..8ced4188b7 100644 --- a/.github/workflows/tests.yml +++ b/.github/workflows/tests.yml @@ -36,7 +36,7 @@ jobs: fail-fast: false matrix: python_version: ["3.10", "3.11"] - pytorch_version: ["2.3.1", "2.4.1", "2.5.0"] + pytorch_version: ["2.3.1", "2.4.1", "2.5.1"] timeout-minutes: 20 steps: @@ -132,7 +132,7 @@ jobs: - cuda: 124 cuda_version: 12.4.1 python_version: "3.11" - pytorch: 2.5.0 + pytorch: 2.5.1 num_gpus: 1 axolotl_extras: steps: