fix import error for triton server(#985)

InternLM · Jan 22, 2024 · 576609f · 576609f
1 parent f456511
commit 576609f
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/lmdeploy/tokenizer.py b/lmdeploy/tokenizer.py
@@ -7,7 +7,10 @@
 
 import torch
 
-from .utils import get_logger
+from lmdeploy.utils import get_logger
+
+# this file will be copied to triton server, make sure all
+# importing are starting from the package root lmdeploy
 
 
 class SentencePieceTokenizer: