Jorge Henao commited on
Commit
6d9f31e
·
1 Parent(s): 04ca423

sentence transformer dimensions fixed

Browse files
Files changed (1) hide show
  1. pinecode_quieries.py +1 -1
pinecode_quieries.py CHANGED
@@ -41,7 +41,7 @@ class PinecodeProposalQueries(DocumentQueries):
41
  #self.retriever = BM25Retriever(document_store = self.document_store)
42
  self.retriever = EmbeddingRetriever(
43
  document_store=self.document_store,
44
- embedding_model="multi-qa-MiniLM-L6-cos-v1",
45
  model_format="sentence_transformers"
46
  )
47
  self.document_store.update_embeddings(self.retriever, batch_size=16)
 
41
  #self.retriever = BM25Retriever(document_store = self.document_store)
42
  self.retriever = EmbeddingRetriever(
43
  document_store=self.document_store,
44
+ embedding_model="multi-qa-MiniLM-L6-dot-v1",
45
  model_format="sentence_transformers"
46
  )
47
  self.document_store.update_embeddings(self.retriever, batch_size=16)