koursaros-ai / nboost

NBoost is a scalable, search-api-boosting platform for deploying transformer models to improve the relevance of search results on different platforms (i.e. Elasticsearch)
Apache License 2.0
674 stars 69 forks source link

Model name 'nboost/pt-tinybert-msmarco' was not found in model name list #62

Closed braaannigan closed 4 years ago

braaannigan commented 4 years ago

I'm trying to follow the example but following either the pip (or docker routes) I get:

nboost                                      --uhost localhost                       
--uport 9200                           
 --search_route "/<index>/_search"      
 --query_path url.query.q               
 --topk_path url.query.size            
  --default_topk 10                      
 --choices_path body.hits.hits           
--cvalues_path _source.passage

Model name 'nboost/pt-tinybert-msmarco' was not found in model name list. This looks similar in some ways to #58 . Any suggestions?

kaykanloo commented 4 years ago

When installing nboost using pip install nboost[pt], for some reason it installs an older version of transformers package (2.2.1). Upgrading that package using the following command solved the issue for me: pip install --upgrade transformers

braaannigan commented 4 years ago

Thanks @kaykanloo, that worked perfectly