From 7c56b6616c3b129ebab079ccb38b8696c64602ac Mon Sep 17 00:00:00 2001 From: ChengZi Date: Mon, 30 Oct 2023 10:13:14 +0800 Subject: [PATCH] fix ernie operator setup Signed-off-by: ChengZi --- src_towhee/pipelines/utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src_towhee/pipelines/utils.py b/src_towhee/pipelines/utils.py index 8399d68..a8dd4c8 100644 --- a/src_towhee/pipelines/utils.py +++ b/src_towhee/pipelines/utils.py @@ -7,7 +7,7 @@ def get_llm_op(config): if config.llm_src.lower() == 'dolly': return ops.LLM.Dolly(model_name=config.dolly_model, **config.llm_kwargs) if config.llm_src.lower() == 'ernie': - return ops.LLM.Ernie(api_key=config.ernie_api_key, secret_key=config.ernie_secret_key, **config.llm_kwargs) + return ops.LLM.Ernie(model_name=config.ernie_model, eb_api_type=config.eb_api_type, eb_access_token=config.eb_access_token, **config.llm_kwargs) if config.llm_src.lower() == 'minimax': return ops.LLM.MiniMax(model=config.minimax_model, api_key=config.minimax_api_key, group_id=config.minimax_group_id, **config.llm_kwargs) if config.llm_src.lower() == 'dashscope':