Update paths

Signed-off-by: Hemil Desai <hemild@nvidia.com>
NVIDIA · Dec 3, 2024 · 3ec4c14 · 3ec4c14
1 parent 426399c
commit 3ec4c14
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tests/functional/test_cases/e2e-nemo2 b/tests/functional/test_cases/e2e-nemo2
@@ -44,9 +44,9 @@ python /opt/NeMo/tests/collections/llm/megatron_gpt_pretraining.py \
   --devices=1 \
   --max-steps=3 \
   --experiment-dir=${PRETRAINING_PATH} \
-  --vocab-path=/datadrive/TestData/nlp/megatron_gpt/data/gpt/vocab.json \
-  --merges-path=/datadrive/TestData/nlp/megatron_gpt/data/gpt/merges.txt \
-  --data-path=/datadrive/TestData/nlp/megatron_gpt/data/gpt/simple_wiki_gpt_preproc_text_document \
+  --vocab-path=${ALIGNER_CI_DIR}/nlp-copy/megatron_gpt/data/gpt/vocab.json \
+  --merges-path=${ALIGNER_CI_DIR}/nlp-copy/megatron_gpt/data/gpt/merges.txt \
+  --data-path=${ALIGNER_CI_DIR}/nlp-copy/megatron_gpt/data/gpt/simple_wiki_gpt_preproc_text_document \
   --index-mapping-dir=tests/collections/llm/gpt_index_mappings \
   --no-masked-softmax-fusion