Move lmdeploy/turbomind/utils.py to lmdeploy/utils.py (#191)

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
 

lmdeploy/turbomind/turbomind.py

@@ -13,7 +13,7 @@ from torch.nn.utils.rnn import pad_sequence
 
 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]