Fix: typo for lmdeploy_wrapper (#171)

* [Fix] Fix typo for lmdeploy_wrapper * [Fix] Remove lmdepoly_wrapper
InternLM · Jun 13, 2024 · ac2190d · ac2190d
1 parent 264ae5c
commit ac2190d
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 2 deletions.
diff --git 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
@@ -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 → lagent/llms/lmdeploy_wrapper.py b/lagent/llms/lmdepoly_wrapper.py → lagent/llms/lmdeploy_wrapper.py