diff --git a/examples/seq2seq/test_seq2seq_examples_multi_gpu.py b/examples/seq2seq/test_seq2seq_examples_multi_gpu.py index efc23b5681..69b979fa01 100644 --- a/examples/seq2seq/test_seq2seq_examples_multi_gpu.py +++ b/examples/seq2seq/test_seq2seq_examples_multi_gpu.py @@ -95,7 +95,7 @@ class TestSummarizationDistillerMultiGPU(TestCasePlus): args = f""" --model_name Helsinki-NLP/opus-mt-en-ro --save_dir {output_dir} - --data_dir test_data/wmt_en_ro + --data_dir {self.test_file_dir_str}/test_data/wmt_en_ro --num_beams 2 --task translation """.split()