diff --git a/examples/internlm2_agent_web_demo.py b/examples/internlm2_agent_web_demo.py index 61f4dccf..ed70b840 100644 --- a/examples/internlm2_agent_web_demo.py +++ b/examples/internlm2_agent_web_demo.py @@ -7,7 +7,7 @@ from lagent.actions import ActionExecutor, ArxivSearch, IPythonInterpreter from lagent.agents.internlm2_agent import INTERPRETER_CN, META_CN, PLUGIN_CN, Internlm2Agent, Internlm2Protocol -from lagent.llms.lmdepoly_wrapper import LMDeployClient +from lagent.llms.lmdeploy_wrapper import LMDeployClient from lagent.llms.meta_template import INTERNLM2_META as META from lagent.schema import AgentStatusCode diff --git a/lagent/llms/__init__.py b/lagent/llms/__init__.py index 75b13eaf..0621addf 100644 --- a/lagent/llms/__init__.py +++ b/lagent/llms/__init__.py @@ -1,7 +1,7 @@ from .base_api import BaseAPIModel from .base_llm import BaseModel from .huggingface import HFTransformer, HFTransformerCasualLM, HFTransformerChat -from .lmdepoly_wrapper import LMDeployClient, LMDeployPipeline, LMDeployServer +from .lmdeploy_wrapper import LMDeployClient, LMDeployPipeline, LMDeployServer from .meta_template import INTERNLM2_META from .openai import GPTAPI from .vllm_wrapper import VllmModel diff --git a/lagent/llms/lmdepoly_wrapper.py b/lagent/llms/lmdeploy_wrapper.py similarity index 100% rename from lagent/llms/lmdepoly_wrapper.py rename to lagent/llms/lmdeploy_wrapper.py