Embedders fix (#133)

code-kern-ai · Nov 27, 2024 · c7ff102 · c7ff102
1 parent b1b565d
commit c7ff102
Show file tree

Hide file tree

Showing 5 changed files with 8 additions and 6 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -103,9 +103,9 @@ httpx==0.27.0
     # via cohere
 httpx-sse==0.4.0
     # via cohere
-huggingface-hub==0.26.2
+huggingface-hub==0.25.2
     # via
-    #   -r requirements/torch-cpu-requirements.txt
+    #   -r requirements/requirements.in
     #   sentence-transformers
     #   tokenizers
     #   transformers

diff --git a/requirements/gpu-requirements.in b/requirements/gpu-requirements.in
@@ -3,4 +3,5 @@ embedders==0.1.8
 spacy==3.7.5
 pydantic==2.7.4
 torchvision==0.17.0 # define version for torchvision to avoid dependency conflict
-openai>=0.27.6,<1.0 # define version for openai to avoid dependency conflict for embedder lib
+openai>=0.27.6,<1.0 # define version for openai to avoid dependency conflict for embedder lib
+huggingface-hub==0.25.2 # define version for huggingface-hub to avoid dependency conflict for embedder lib
diff --git a/requirements/requirements.in b/requirements/requirements.in
@@ -4,4 +4,5 @@ spacy==3.7.5
 pydantic==2.7.4
 torchvision==0.17.0 # define version for torchvision to avoid dependency conflict
 sentence-transformers~=2.2.0 # last version with default_prompt_name & pooling_mode_weightedmean_tokens # higher only possible with embedders/gates change
-openai>=0.27.6,<1.0 # define version for openai to avoid dependency conflict for embedder lib
+openai>=0.27.6,<1.0 # define version for openai to avoid dependency conflict for embedder lib
+huggingface-hub==0.25.2 # define version for huggingface-hub to avoid dependency conflict for embedder lib
diff --git a/requirements/torch-cpu-requirements.txt b/requirements/torch-cpu-requirements.txt
@@ -39,7 +39,7 @@ fsspec==2024.10.0
     #   torch
 h11==0.14.0
     # via uvicorn
-huggingface-hub==0.26.2
+# huggingface-hub==0.26.2
     # via
     #   tokenizers
     #   transformers

diff --git a/requirements/torch-cuda-requirements.txt b/requirements/torch-cuda-requirements.txt
@@ -39,7 +39,7 @@ fsspec==2024.10.0
     #   torch
 h11==0.14.0
     # via uvicorn
-huggingface-hub==0.26.2
+# huggingface-hub==0.26.2
     # via
     #   tokenizers
     #   transformers