update to use tmp in ci tests

Signed-off-by: yaoyu-33 <[email protected]>
NVIDIA · Nov 22, 2024 · 8344cc9 · 8344cc9
1 parent 617c8ca
commit 8344cc9
Showing 1 changed file with 2 additions and 6 deletions.
diff --git a/.github/workflows/cicd-main.yml b/.github/workflows/cicd-main.yml
@@ -3929,9 +3929,7 @@ jobs:
         python tests/collections/vlm/neva_train.py \
         --devices=1 \
         --max-steps=5 \
-        --experiment-dir=tests/collections/vlm/neva_results/${{ github.run_id }} \
-      AFTER_SCRIPT: |
-        rm -rf tests/collections/vlm/neva_results/${{ github.run_id }}
+        --experiment-dir=/tmp/nemo2_neva_results/${{ github.run_id }}
 
   L2_NeMo_2_MLLAMA_MOCK_TRAINING:
     needs: [cicd-test-container-setup]
@@ -3943,9 +3941,7 @@ jobs:
         python tests/collections/vlm/mllama_train.py \
         --devices=1 \
         --max-steps=5 \
-        --experiment-dir=tests/collections/vlm/mllama_results/${{ github.run_id }} \
-      AFTER_SCRIPT: |
-        rm -rf tests/collections/vlm/mllama_results/${{ github.run_id }}
+        --experiment-dir=/tmp/nemo2_mllama_results/${{ github.run_id }}
 
   L2_NeMo_2_Mixtral_Pretraining:
       needs: [cicd-test-container-setup]