Skip to content

Commit

Permalink
Add MessagePack serialization as a top level dependency, closes #771
Browse files Browse the repository at this point in the history
  • Loading branch information
davidmezzetti committed Aug 24, 2024
1 parent e473379 commit d30d298
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -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", "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.28.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"]
Expand Down Expand Up @@ -37,7 +37,6 @@
extras["api"] = [
"aiohttp>=3.8.1",
"fastapi>=0.94.0",
"msgpack>=1.0.7",
"pillow>=7.1.2",
"python-multipart>=0.0.7",
"uvicorn>=0.12.1",
Expand Down Expand Up @@ -91,6 +90,7 @@
"pymagnitude-lite>=0.1.43",
"scikit-learn>=0.23.1",
"sentence-transformers>=2.2.0",
"skops>=0.9.0",
]

extras["workflow"] = [
Expand Down

0 comments on commit d30d298

Please sign in to comment.