Skip to content

GitLab

Unverified Commit 1eb40338 authored by Stas Bekman's avatar Stas Bekman Committed by GitHub
Browse files

[deepspeed tests] fix summarization (#15149)

parent 6e058e84
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
tests/deepspeed/test_model_zoo.py
View file @ 1eb40338
...@@ -157,6 +157,7 @@ def make_task_cmds(): ...@@ -157,6 +157,7 @@ def make_task_cmds():
--train_file {data_dir_xsum}/sample.json --train_file {data_dir_xsum}/sample.json
--max_source_length 12 --max_source_length 12
--max_target_length 12 --max_target_length 12
--lang en
""", """,
clm=f""" clm=f"""
{scripts_dir}/language-modeling/run_clm.py {scripts_dir}/language-modeling/run_clm.py
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment