From d55d6da22d9e2072f143fc05825f645647264411 Mon Sep 17 00:00:00 2001 From: Farshid Zavareh Date: Mon, 28 Oct 2024 16:05:16 +1100 Subject: [PATCH] Add query prefix for Stella base model (#1023) --- src/marqo/marqo_docs.py | 3 +-- src/marqo/s2_inference/model_registry.py | 3 ++- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/src/marqo/marqo_docs.py b/src/marqo/marqo_docs.py index 223078cbb..b6e135619 100644 --- a/src/marqo/marqo_docs.py +++ b/src/marqo/marqo_docs.py @@ -78,5 +78,4 @@ def search_api_score_modifiers_parameter(): def hugging_face_trust_remote_code(): - # TODO - Update to valid docs URL - return _build_url('reference/api/models/hugging-face-models/#trust-remote-code') + return _build_url('models/marqo/bring-your-own-model/#bring-your-own-hugging-face-sentence-transformers-models') diff --git a/src/marqo/s2_inference/model_registry.py b/src/marqo/s2_inference/model_registry.py index 056945efb..502f9de55 100644 --- a/src/marqo/s2_inference/model_registry.py +++ b/src/marqo/s2_inference/model_registry.py @@ -900,7 +900,8 @@ def _get_hf_properties() -> Dict: "dimensions": 1024, "tokens": 512, "type": "hf_stella", - "trustRemoteCode": True + "trustRemoteCode": True, + "text_query_prefix": "Instruct: Given a web search query, retrieve relevant passages that answer the query.\nQuery: " }, } return HF_MODEL_PROPERTIES