yujunjun/haystack
1
0
Fork 0
mirror of https://github.com/deepset-ai/haystack.git synced 2025-09-04 13:53:16 +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

6
haystack/retriever/dense.py
View File

@ -298,8 +298,12 @@ class EmbeddingRetriever(BaseRetriever):
) )
elif model_format == "sentence_transformers": elif model_format == "sentence_transformers":
try:
from sentence_transformers import SentenceTransformer 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 # pretrained embedding models coming from: https://github.com/UKPLab/sentence-transformers#pretrained-models
# e.g. 'roberta-base-nli-stsb-mean-tokens' # e.g. 'roberta-base-nli-stsb-mean-tokens'
if use_gpu: if use_gpu: