Spaces:
Sleeping
Sleeping
Temporal fix for incompatibility between langchain_huggingface and sentence-transformers<2.6
Browse files- requirements.txt +0 -1
- 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 |
|