diff --git a/setup.py b/setup.py index 5307a4c2c..2e5e9fc66 100644 --- a/setup.py +++ b/setup.py @@ -6,7 +6,7 @@ DESCRIPTION = "".join([line for line in f if "gh-dark-mode-only" not in line]) # Required dependencies -install = ["faiss-cpu>=1.7.1.post2", "msgpack>=1.0.7", "torch>=1.12.1", "transformers>=4.28.0"] +install = ["faiss-cpu>=1.7.1.post2", "msgpack>=1.0.7", "torch>=1.12.1", "transformers>=4.45.0"] # Required dependencies that are also transformers dependencies install += ["huggingface-hub>=0.19.0", "numpy>=1.18.4", "pyyaml>=5.3", "regex>=2022.8.17"]