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