diff --git a/modelscope_agent/llm/__init__.py b/modelscope_agent/llm/__init__.py index 50b0b043..2e036570 100644 --- a/modelscope_agent/llm/__init__.py +++ b/modelscope_agent/llm/__init__.py @@ -17,7 +17,7 @@ def get_chat_model(model: str, model_server: str, **kwargs) -> BaseChatModel: """ model_type = re.split(r'[-/_]', model)[0] # parser qwen / gpt / ... registered_model_id = f'{model_server}_{model_type}' - print(LLM_REGISTRY) + if registered_model_id in LLM_REGISTRY: # specific model from specific source return LLM_REGISTRY[registered_model_id](model, model_server, **kwargs) elif model_server in LLM_REGISTRY: # specific source