From 61ddac3259d4a0ec18349db845efbb19b27f32c1 Mon Sep 17 00:00:00 2001 From: "junjie.jiang" Date: Wed, 7 Jun 2023 16:39:47 +0800 Subject: [PATCH] Update Signed-off-by: junjie.jiang --- eqa_search.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/eqa_search.py b/eqa_search.py index 1ccd9a2..7cd3843 100644 --- a/eqa_search.py +++ b/eqa_search.py @@ -89,10 +89,10 @@ def _get_llm_op(config): return ops.LLM.OpenAI(model_name=config.openai_model, api_key=config.openai_api_key) if config.llm_src.lower() == 'dolly': return ops.LLM.Dolly(model_name=config.dolly_model) - if config.llm_src.lower() == ' ernie': + if config.llm_src.lower() == 'ernie': return ops.LLM.Ernie(api_key=config.ernie_api_key, secret_key=config.ernie_secret_key) - raise RuntimeError('Unknown llm source: [%s], only support \'openai\' and \'dolly\'' % (config.llm_src)) + raise RuntimeError('Unknown llm source: [%s], only support openai, dolly and ernie' % (config.llm_src)) def _get_prompt(config):