fix ernie operator setup (#92)

Signed-off-by: ChengZi <[email protected]>
zilliztech · Oct 30, 2023 · 4aaabb6 · 4aaabb6
1 parent f107408
commit 4aaabb6
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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':