klbr/khoj
1
0
Fork 0
mirror of https://github.com/khoaliber/khoj.git synced 2026-03-05 21:29:11 +00:00
Code Issues Packages Projects Releases Wiki Activity

Upgrade sentence-transformers. Avoid transformers v4.50.0 as problematic

Browse Source 
- The 3.4.1 release of sentence tranformer fixes offline load latency
  of sentence transformer models (and Khoj) by avoiding call to HF

- The 4.50.0 release of transformers is resulting in
  jax error (unexpected keyword argument 'flatten_with_keys') on load.
This commit is contained in:
Debanjum
2025-03-23 08:38:29 +05:30
parent 510cbed61c
commit f2b438145f
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
pyproject.toml
View File

@@ -51,9 +51,9 @@ dependencies = [
"pyyaml ~= 6.0",
"rich >= 13.3.1",
"schedule == 1.1.0",
"sentence-transformers == 3.0.1",
"sentence-transformers == 3.4.1",
"einops == 0.8.0",
"transformers >= 4.28.0",
"transformers >= 4.28.0, < 4.50.0",
"torch == 2.2.2",
"uvicorn == 0.30.6",
"aiohttp ~= 3.9.0",