diff --git a/modelscope/models/nlp/llama/__init__.py b/modelscope/models/nlp/llama/__init__.py
index d5b6fd19..848bad02 100644
--- a/modelscope/models/nlp/llama/__init__.py
+++ b/modelscope/models/nlp/llama/__init__.py
@@ -1,8 +1,11 @@
 # Copyright (c) Alibaba, Inc. and its affiliates.
 from typing import TYPE_CHECKING
 
-from transformers.models.llama import (LlamaConfig, LlamaTokenizer,
-                                       LlamaTokenizerFast)
+# from transformers.models.llama import (LlamaConfig, LlamaTokenizer,
+#                                        LlamaTokenizerFast)
+
+from transformers import LlamaTokenizer
+from transformers.models.llama import (LlamaConfig, LlamaTokenizerFast)
 
 from modelscope.utils.import_utils import LazyImportModule