Install sentencepiece in DeepSpeed CI image (#20795)

* Install sentencepiece in DS CI image * update Co-authored-by: ydshieh <[email protected]>
huggingface · Dec 16, 2022 · 7032e02 · 7032e02
1 parent 26dd041
commit 7032e02
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -317,7 +317,7 @@ def run(self):
     + extras["modelcreation"]
 )
 
-extras["deepspeed-testing"] = extras["deepspeed"] + extras["testing"] + extras["optuna"]
+extras["deepspeed-testing"] = extras["deepspeed"] + extras["testing"] + extras["optuna"] + extras["sentencepiece"]
 
 extras["quality"] = deps_list("black", "datasets", "isort", "flake8", "GitPython", "hf-doc-builder")