Update _test_upstream_pax.yaml

Disabled saving hlo by default as suggested by terry
NVIDIA · May 20, 2024 · da69dbd · da69dbd
1 parent f44cdef
commit da69dbd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/.github/workflows/_test_upstream_pax.yaml b/.github/workflows/_test_upstream_pax.yaml
@@ -220,7 +220,7 @@ jobs:
             PARALLEL_CONFIG: [1, 1, 8, 1]
             BATCH_SIZE: 4
             EVALUATE: true
-            ADDITIONAL_ARGS: "--model-type LLaMA70BProxy --evaluate --save-hlo 0"
+            ADDITIONAL_ARGS: "--model-type LLaMA70BProxy --evaluate"
           - TEST_NAME: 2DP1FSDP1TP4PP_fmha
             PARALLEL_CONFIG: [4, 2, 1, 1]
             BATCH_SIZE: 4