From 3ec4c1415d9dac317881d0bc4c1fff5af8b9e082 Mon Sep 17 00:00:00 2001 From: Hemil Desai Date: Mon, 2 Dec 2024 14:56:13 -0800 Subject: [PATCH] Update paths Signed-off-by: Hemil Desai --- tests/functional/test_cases/e2e-nemo2 | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/functional/test_cases/e2e-nemo2 b/tests/functional/test_cases/e2e-nemo2 index d56d7bae6..a51a0360e 100755 --- 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