Use llama_ci for now as megatron_amp_O2_hf_tokenizer model needs to b…

…e created Signed-off-by: Jan Lasek <janek.lasek@gmail.com>
NVIDIA · Apr 26, 2024 · c30fe70 · c30fe70
1 parent 742799d
commit c30fe70
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/.github/workflows/cicd-main.yml b/.github/workflows/cicd-main.yml
@@ -418,7 +418,7 @@ jobs:
           uses: actions/checkout@v2
         - run: |
             python examples/nlp/language_modeling/megatron_llama_quantization.py \
-            model_file=/home/TestData/nlp/megatron_llama/llama_ci_megatron_amp_O2_hf_tokenizer.nemo \
+            model_file=/home/TestData/nlp/megatron_llama/llama_ci.nemo \
             quantization.calib_dataset=/home/TestData/nlp/test_quantization/test.json \
             quantization.algorithm=int8_sq \
             quantization.num_calib_size=8 \