move CUDA_VERSION_BNB arg inside of stage build scope

2023-05-29 13:30:15 -04:00
parent 00323f0a6f
commit e43bcc6c4f
2 changed files with 3 additions and 5 deletions
--- a/.github/workflows/base.yml
+++ b/.github/workflows/base.yml
@@ -16,11 +16,11 @@ jobs:
        include:
          - cuda: cu118
            cuda_version: 11.8.0
-            cuda_version_bnb: 118
+            cuda_version_bnb: "118"
            pytorch: 2.0.0
          - cuda: cu117
            cuda_version: 11.7.0
-            cuda_version_bnb: 117
+            cuda_version_bnb: "117"
            pytorch: 1.13.1
    steps:
      - name: Checkout