[deepspeed tests] fix summarization (#15149)

1eb40338 · Stas Bekman · GitHub · 6e058e84 · 1eb40338
Unverified Commit 1eb40338 authored Jan 13, 2022 by Stas Bekman Committed by GitHub Jan 13, 2022
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

tests/deepspeed/test_model_zoo.py tests/deepspeed/test_model_zoo.py +1 -0

No files found.
--- 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