Merge branch 'main' into packing-resumption

mosaicml · Jun 14, 2024 · 33f84cf · 33f84cf
2 parents 4d53147 + 82ef072
commit 33f84cf
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/.github/workflows/pr-gpu.yaml b/.github/workflows/pr-gpu.yaml
@@ -25,7 +25,7 @@ jobs:
           pytest_command: "coverage run -m pytest"
           pip_deps: "[all]"
         - name: "gpu-2.3.1"
-          container: mosaicml/llm-foundry:2.3.1_cu121_flash2-latest
+          container: mosaicml/llm-foundry:2.3.1_cu121-latest
           markers: "gpu"
           pytest_command: "coverage run -m pytest"
           pip_deps: "[all]"