From 4aaabb6badf4efe40f53affc033808a6113e39c1 Mon Sep 17 00:00:00 2001 From: ChengZi Date: Mon, 30 Oct 2023 11:12:03 +0800 Subject: [PATCH] fix ernie operator setup (#92) Signed-off-by: ChengZi --- src_towhee/pipelines/utils.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src_towhee/pipelines/utils.py b/src_towhee/pipelines/utils.py index 8399d68..bf73e30 100644 --- a/src_towhee/pipelines/utils.py +++ b/src_towhee/pipelines/utils.py @@ -7,7 +7,8 @@ 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':