clean up

allenai · Oct 31, 2024 · af4fa46 · af4fa46
1 parent dd2ca7f
commit af4fa46
Showing 1 changed file with 5 additions and 6 deletions.
diff --git a/.github/workflows/main.yml b/.github/workflows/main.yml
@@ -109,7 +109,7 @@ jobs:
       matrix:
         task:
           - name: Test (GPU)
-            image: olmo-core-dev
+            image: olmo-core
             gpus: 2
             run: |
               pytest -v --color=yes --durations=3 -m gpu \
@@ -118,14 +118,14 @@ jobs:
                 src/test/
 
           - name: Test checkpoint (GPU)
-            image: olmo-core-dev
+            image: olmo-core
             gpus: 2
             run: |
               pytest -v --color=yes --durations=3 -m gpu \
                 src/test/distributed/checkpoint*
 
           - name: Test MoE (GPU)
-            image: olmo-core-dev
+            image: olmo-core
             gpus: 1
             run: |
               pytest -v --color=yes --durations=3 -m gpu \
@@ -185,7 +185,7 @@ jobs:
                     - ai2/pluto-cirrascale
                     # - ai2/allennlp-cirrascale
                     # - ai2/allennlp-elanding-a100-40g
-                    # - ai2/saturn-cirrascale
+                    - ai2/saturn-cirrascale
                 envVars:
                   - name: CUBLAS_WORKSPACE_CONFIG
                     value: ":16:8"
@@ -202,8 +202,7 @@ jobs:
                 result:
                   path: /unused
           token: ${{ env.BEAKER_TOKEN }}
-          # workspace: ${{ env.BEAKER_WORKSPACE }}
-          workspace: ai2/OLMo-pretraining-stability
+          workspace: ${{ env.BEAKER_WORKSPACE }}
 
   release:
     name: Release