Upgrade sentence-transformer package to version 3.0.1

Add einops dependency for some sentence transformer models like the
nomic-embed
This commit is contained in:
Debanjum Singh Solanky
2024-07-06 18:44:38 +05:30
parent d693baccbc
commit 4a471979eb

View File

@@ -52,7 +52,8 @@ dependencies = [
"pyyaml ~= 6.0",
"rich >= 13.3.1",
"schedule == 1.1.0",
"sentence-transformers == 2.5.1",
"sentence-transformers == 3.0.1",
"einops == 0.8.0",
"transformers >= 4.28.0",
"torch == 2.2.2",
"uvicorn == 0.17.6",