Install sentencepiece in DeepSpeed CI image (#20795)
* Install sentencepiece in DS CI image * update Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
This commit is contained in:
2
setup.py
2
setup.py
@@ -317,7 +317,7 @@ extras["testing"] = (
|
|||||||
+ extras["modelcreation"]
|
+ 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")
|
extras["quality"] = deps_list("black", "datasets", "isort", "flake8", "GitPython", "hf-doc-builder")
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user