diff --git a/setup.py b/setup.py index e16e1ed253b0..4d4d38fc4d32 100644 --- a/setup.py +++ b/setup.py @@ -98,7 +98,7 @@ extras["serving"] = ["pydantic", "uvicorn", "fastapi", "starlette"] -extras["sentencepiece"] = ["sentencepiece!=0.1.92"] +extras["sentencepiece"] = ["sentencepiece==0.1.91"] extras["retrieval"] = ["faiss-cpu", "datasets"] extras["testing"] = ["pytest", "pytest-xdist", "timeout-decorator", "parameterized", "psutil"] + extras["retrieval"] # sphinx-rtd-theme==0.5.0 introduced big changes in the style. @@ -140,7 +140,7 @@ # for OpenAI GPT "regex != 2019.12.17", # for SentencePiece models - "sentencepiece != 0.1.92", + "sentencepiece == 0.1.91", "protobuf", # for XLM "sacremoses",