From 1eb40338ac41a2ffbb2137216de3bb63bb739aad Mon Sep 17 00:00:00 2001
From: Stas Bekman <stas00@users.noreply.github.com>
Date: Thu, 13 Jan 2022 13:48:51 -0800
Subject: [PATCH] [deepspeed tests] fix summarization (#15149)

---
 tests/deepspeed/test_model_zoo.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/tests/deepspeed/test_model_zoo.py b/tests/deepspeed/test_model_zoo.py
index 321e8b2bf0..f2b04abab7 100644
--- 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