From 7032e0203262ebb2ebf55da8d2e01f873973e835 Mon Sep 17 00:00:00 2001 From: Yih-Dar <2521628+ydshieh@users.noreply.github.com> Date: Fri, 16 Dec 2022 18:23:46 +0100 Subject: [PATCH] Install `sentencepiece` in `DeepSpeed` CI image (#20795) * Install sentencepiece in DS CI image * update Co-authored-by: ydshieh --- setup.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/setup.py b/setup.py index b089b561f1cd..91ee3642024f 100644 --- 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")