Skip to content

Commit fd40440

Browse files
committed
fix
Signed-off-by: Chen-0210 <chenjincong11@gmail.com>
1 parent 0f2f64a commit fd40440

File tree

1 file changed

+4
-5
lines changed

1 file changed

+4
-5
lines changed

vllm/transformers_utils/config.py

+4-5
Original file line numberDiff line numberDiff line change
@@ -18,11 +18,6 @@
1818
from vllm.transformers_utils.utils import check_gguf_file
1919
from vllm.utils import resolve_obj_by_qualname
2020

21-
if VLLM_USE_MODELSCOPE:
22-
from modelscope import AutoConfig
23-
else:
24-
from transformers import AutoConfig
25-
2621
MISTRAL_CONFIG_NAME = "params.json"
2722
HF_TOKEN = os.getenv('HF_TOKEN', None)
2823

@@ -294,6 +289,10 @@ def get_config(
294289
)
295290
else:
296291
try:
292+
if VLLM_USE_MODELSCOPE:
293+
from modelscope import AutoConfig
294+
else:
295+
from transformers import AutoConfig
297296
config = AutoConfig.from_pretrained(
298297
model,
299298
trust_remote_code=trust_remote_code,

0 commit comments

Comments
 (0)