From 82ef0725515e0c97d7fafe1aa5c1abe9120295d1 Mon Sep 17 00:00:00 2001
From: Daniel King <43149077+dakinggg@users.noreply.github.com>
Date: Fri, 14 Jun 2024 11:56:53 -0700
Subject: [PATCH] fix typo in CI (#1284)

---
 .github/workflows/pr-gpu.yaml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/.github/workflows/pr-gpu.yaml b/.github/workflows/pr-gpu.yaml
index 335d049306..07e811e244 100644
--- 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]"