Update error msg

Signed-off-by: Kaiyuan Hu <kaiyuan.hu@zilliz.com>
2 years ago · a1c312b28d
1 changed files with 1 additions and 1 deletions
--- a/eqa_search.py
+++ b/eqa_search.py
@ -67,7 +67,7 @@ def _get_embedding_op(config):
            model_name=config.embedding_model, api_key=config.openai_api_key
        )

-    raise RuntimeError('Unknown model: [%s], only support: %s' % (config.embedding_model, _hf_models + _openai_models))
+    raise RuntimeError('Unknown model: [%s], only support: %s' % (config.embedding_model, _hf_models + _sbert_models + _openai_models))


 def _get_similarity_evaluation_op(config):