同步0.2.6代码

lmdeploy/cli/__init__.py

 # Copyright (c) OpenMMLab. All rights reserved.
-from .cli import run
+from .entrypoint import run
 
 __all__ = ['run']

lmdeploy/lite/utils/load.py

@@ -3,7 +3,7 @@
 import torch
 from transformers import AutoConfig, AutoModelForCausalLM
 
-from lmdeploy.pytorch.model import LoadWoInit
+from lmdeploy.pytorch.accel import LoadNoInit
 
 
 def load_hf_from_pretrained(pretrained_model_name_or_path,
@@ -26,7 +26,7 @@ def load_hf_from_pretrained(pretrained_model_name_or_path,
     elif dtype == torch.bfloat16:
         hf_config.bf16 = True
 
-    with LoadWoInit():
+    with LoadNoInit():
         # Load model
         model = AutoModelForCausalLM.from_pretrained(
             pretrained_model_name_or_path, config=hf_config, **kwargs)