add torch 2.3.0 to builds (#1593)

2024-05-05 18:45:45 -04:00
parent 120b809465
commit 70185763f6
3 changed files with 25 additions and 0 deletions
--- a/.github/workflows/main.yml
+++ b/.github/workflows/main.yml
@@ -30,6 +30,11 @@ jobs:
            python_version: "3.11"
            pytorch: 2.2.1
            axolotl_extras:
+          - cuda: 121
+            cuda_version: 12.1.0
+            python_version: "3.11"
+            pytorch: 2.3.0
+            axolotl_extras:
    runs-on: axolotl-gpu-runner
    steps:
      - name: Checkout
@@ -86,6 +91,11 @@ jobs:
            python_version: "3.11"
            pytorch: 2.2.1
            axolotl_extras:
+          - cuda: 121
+            cuda_version: 12.1.0
+            python_version: "3.11"
+            pytorch: 2.3.0
+            axolotl_extras:
    runs-on: axolotl-gpu-runner
    steps:
      - name: Checkout