fix typo in CI (#1284)

mosaicml · Jun 14, 2024 · 82ef072 · 82ef072
1 parent 1a2fac0
commit 82ef072
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]"