[misc] error early for old-style class (#10304)

Signed-off-by: youkaichao <youkaichao@gmail.com>
This commit is contained in:
youkaichao
2024-11-13 18:55:39 -08:00
committed by GitHub
parent 15bb8330aa
commit 504ac53d18
21 changed files with 75 additions and 63 deletions

View File

@@ -14,7 +14,6 @@ from vllm.attention.selector import (_Backend, backend_name_to_enum,
from vllm.config import VllmConfig
from vllm.logger import init_logger
from vllm.model_executor.model_loader.weight_utils import default_weight_loader
from vllm.model_executor.models import ModelRegistry
from vllm.multimodal import MultiModalPlaceholderMap, NestedTensors
from vllm.platforms import current_platform
from vllm.sequence import IntermediateTensors
@@ -240,12 +239,9 @@ def init_vllm_registered_model(
Helper function to initialize an inner model registered to vLLM,
based on the arguments passed to the outer vLLM model.
"""
model_class, _ = ModelRegistry.resolve_model_cls(hf_config.architectures)
return model_class(
vllm_config=vllm_config.with_hf_config(hf_config),
prefix=prefix,
)
from vllm.model_executor.model_loader.loader import _initialize_model
vllm_config = vllm_config.with_hf_config(hf_config)
return _initialize_model(vllm_config, prefix)
@overload