[deepspeed tests] fix summarization (#15149)

2022-01-13 13:48:51 -08:00
parent 6e058e84fd
commit 1eb40338ac
1 changed files with 1 additions and 0 deletions
--- 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