make the kd e2e fit in vram for ci and add lora version

2025-01-08 11:07:29 -05:00
parent 1d039f5486
commit 432f65f5e6
2 changed files with 38 additions and 6 deletions
--- a/.github/workflows/tests.yml
+++ b/.github/workflows/tests.yml
@@ -207,7 +207,7 @@ jobs:
          - cuda: 124
            cuda_version: 12.4.1
            python_version: "3.11"
-            pytorch: 2.4.1
+            pytorch: 2.5.1
            num_gpus: 1
            axolotl_extras:
    steps:
@@ -253,7 +253,7 @@ jobs:
          - cuda: 124
            cuda_version: 12.4.1
            python_version: "3.11"
-            pytorch: 2.5.1
+            pytorch: 2.4.1
            num_gpus: 1
            axolotl_extras:
    steps: