Skip to content

Commit

Permalink
fix ernie operator setup
Browse files Browse the repository at this point in the history
Signed-off-by: ChengZi <[email protected]>
  • Loading branch information
zc277584121 committed Oct 30, 2023
1 parent f107408 commit fa896bd
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src_towhee/pipelines/utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -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, api_key=config.eb_api_type, secret_key=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':
Expand Down

0 comments on commit fa896bd

Please sign in to comment.