From a1c312b28d1123574c689a0e44a86170bace226f Mon Sep 17 00:00:00 2001 From: Kaiyuan Hu Date: Wed, 31 May 2023 11:37:51 +0800 Subject: [PATCH] Update error msg Signed-off-by: Kaiyuan Hu --- eqa_search.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/eqa_search.py b/eqa_search.py index e5e27f9..0786104 100644 --- 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):