PabloVD commited on
Commit
2da0c1f
·
1 Parent(s): ac138df

Temporal fix for incompatibility between langchain_huggingface and sentence-transformers<2.6

Browse files
Files changed (2) hide show
  1. requirements.txt +0 -1
  2. worker.py +11 -0
requirements.txt CHANGED
@@ -2,4 +2,3 @@ langchain
2
  langchain-community
3
  langchain-huggingface
4
  chromadb
5
- sentence-transformers
 
2
  langchain-community
3
  langchain-huggingface
4
  chromadb
 
worker.py CHANGED
@@ -7,6 +7,17 @@ from langchain.text_splitter import RecursiveCharacterTextSplitter
7
  from langchain_community.vectorstores import Chroma
8
  from langchain_huggingface import HuggingFaceEndpoint
9
 
 
 
 
 
 
 
 
 
 
 
 
10
  # Check for GPU availability and set the appropriate device for computation.
11
  DEVICE = "cuda:0" if torch.cuda.is_available() else "cpu"
12
 
 
7
  from langchain_community.vectorstores import Chroma
8
  from langchain_huggingface import HuggingFaceEndpoint
9
 
10
+ import pip
11
+
12
+ def install(package):
13
+ if hasattr(pip, 'main'):
14
+ pip.main(['install', package])
15
+ else:
16
+ pip._internal.main(['install', package])
17
+
18
+ # Temporal fix for incompatibility between langchain_huggingface and sentence-transformers<2.6
19
+ install("sentence-transformers==2.2.2")
20
+
21
  # Check for GPU availability and set the appropriate device for computation.
22
  DEVICE = "cuda:0" if torch.cuda.is_available() else "cpu"
23