yujunjun/haystack
1
0
Fork 0
mirror of https://github.com/deepset-ai/haystack.git synced 2025-09-03 13:23:38 +00:00
Code Issues Packages Projects Releases Wiki Activity

Make sentence-transformers usage more user-friendly (#439)

Browse Source 
Co-authored-by: guillim <guigloo@msn.com>
This commit is contained in:
Malte Pietsch 2020-09-28 15:34:23 +02:00 committed by GitHub
parent cd19d65f1a
commit bb4802ae6a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
haystack/retriever/dense.py
View File

@ -298,8 +298,12 @@ class EmbeddingRetriever(BaseRetriever):
)
elif model_format == "sentence_transformers":
from sentence_transformers import SentenceTransformer
try:
from sentence_transformers import SentenceTransformer
except ImportError:
raise ImportError("Can't find package `sentence-transformers` \n"
"You can install it via `pip install sentece-transformers` \n"
"For details see https://github.com/UKPLab/sentence-transformers ")
# pretrained embedding models coming from: https://github.com/UKPLab/sentence-transformers#pretrained-models
# e.g. 'roberta-base-nli-stsb-mean-tokens'
if use_gpu: