feat: add torch2.6 to ci (#2311)

2025-02-07 19:28:54 +07:00
parent a971eb4ce6
commit 5bbad5ef93
7 changed files with 45 additions and 8 deletions
--- a/.github/workflows/multi-gpu-e2e.yml
+++ b/.github/workflows/multi-gpu-e2e.yml
@@ -34,6 +34,13 @@ jobs:
            axolotl_extras:
            num_gpus: 2
            nightly_build: "true"
+          - cuda: 124
+            cuda_version: 12.4.1
+            python_version: "3.11"
+            pytorch: 2.6.0
+            axolotl_extras:
+            num_gpus: 2
+            nightly_build: "true"
    runs-on: [self-hosted, modal]
    timeout-minutes: 120
    steps: