move lmdeploy/turbomind/utils.py to lmdeploy/utils.py

lvhan028 · Aug 2, 2023 · 3e1e154 · 3e1e154
1 parent 4767b04
commit 3e1e154
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 2 deletions.
diff --git a/lmdeploy/serve/turbomind/chatbot.py b/lmdeploy/serve/turbomind/chatbot.py
@@ -52,7 +52,7 @@ def stream_callback(que, result, error):
 
 def get_logger(log_file=None, log_level=logging.INFO):
     """Return the logger."""
-    from lmdeploy.turbomind.utils import get_logger
+    from lmdeploy.utils import get_logger
     logger = get_logger('service.ft', log_file=log_file, log_level=log_level)
     return logger
 

diff --git a/lmdeploy/turbomind/turbomind.py b/lmdeploy/turbomind/turbomind.py
@@ -13,7 +13,7 @@
 
 import lmdeploy
 from lmdeploy.model import MODELS
-from lmdeploy.turbomind.utils import get_logger
+from lmdeploy.utils import get_logger
 
 # TODO: find another way import _turbomind
 lmdeploy_dir = osp.split(lmdeploy.__file__)[0]

diff --git a/lmdeploy/turbomind/utils.py → lmdeploy/utils.py b/lmdeploy/turbomind/utils.py → lmdeploy/utils.py