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):