[deepspeed tests] fix summarization (huggingface#15149)

shen9712 · Jan 13, 2022 · 1eb4033 · 1eb4033
1 parent 6e058e8
commit 1eb4033
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tests/deepspeed/test_model_zoo.py b/tests/deepspeed/test_model_zoo.py
@@ -157,6 +157,7 @@ def make_task_cmds():
         --train_file {data_dir_xsum}/sample.json
         --max_source_length 12
         --max_target_length 12
+        --lang en
         """,
         clm=f"""
         {scripts_dir}/language-modeling/run_clm.py