From 3b81fafa3e019af838a1339797754752706cb380 Mon Sep 17 00:00:00 2001 From: Debanjum Singh Solanky Date: Sun, 15 Aug 2021 23:57:22 -0700 Subject: [PATCH] Use updated path to MiniLM bi-encoder model on hugging-face --- search_types/asymmetric.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/search_types/asymmetric.py b/search_types/asymmetric.py index 0b0ddcc3..ff191b11 100644 --- a/search_types/asymmetric.py +++ b/search_types/asymmetric.py @@ -14,9 +14,9 @@ import pathlib def initialize_model(): "Initialize model for assymetric semantic search. That is, where query smaller than results" - bi_encoder = SentenceTransformer('msmarco-MiniLM-L-6-v3') # The bi-encoder encodes all entries to use for semantic search - top_k = 100 # Number of entries we want to retrieve with the bi-encoder - cross_encoder = CrossEncoder('cross-encoder/ms-marco-MiniLM-L-6-v2') # The cross-encoder re-ranks the results to improve quality + bi_encoder = SentenceTransformer('sentence-transformers/msmarco-MiniLM-L-6-v3') # The bi-encoder encodes all entries to use for semantic search + top_k = 100 # Number of entries we want to retrieve with the bi-encoder + cross_encoder = CrossEncoder('cross-encoder/ms-marco-MiniLM-L-6-v2') # The cross-encoder re-ranks the results to improve quality return bi_encoder, cross_encoder, top_k