From 4151179810f97b641b86be1d121ca9910a5df1e3 Mon Sep 17 00:00:00 2001
From: epwalsh <epwalsh10@gmail.com>
Date: Mon, 20 Mar 2023 14:55:52 -0700
Subject: [PATCH] fix

---
 .github/workflows/main.yml | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/.github/workflows/main.yml b/.github/workflows/main.yml
index e3dc82a96..95990c57f 100644
--- a/.github/workflows/main.yml
+++ b/.github/workflows/main.yml
@@ -110,8 +110,6 @@ jobs:
       BEAKER_TOKEN: ${{ secrets.BEAKER_TOKEN }}
       BEAKER_IMAGE: dolma-test
       BEAKER_WORKSPACE: ai2/llm-testing
-      # For reproducibility
-      CUBLAS_WORKSPACE_CONFIG: ':16:8'
     steps:
       - name: Determine current commit SHA (pull request)
         if: github.event_name == 'pull_request'
@@ -155,6 +153,8 @@ jobs:
                     value: ${{ secrets.GITHUB_TOKEN }}
                   - name: CUDA_LAUNCH_BLOCKING
                     value: "1"
+                  - name: CUBLAS_WORKSPACE_CONFIG
+                    value: ":16:8"
                   - name: TOKENIZERS_PARALLELISM
                     value: "false"
                 command: ["/entrypoint.sh", "pytest", "-v", "-m", "gpu", "tests/"]