Make it optional to set the encoder, cross-encoder configs via admin UI

Upgrade sentence-transformer, add einops dependency for some sentence
transformer models like nomic
This commit is contained in:
Debanjum Singh Solanky
2024-07-05 16:07:42 +05:30
parent 6d59ad7fc9
commit a353d883a0
2 changed files with 5 additions and 4 deletions

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",