From 03c782eb91f70ba990432aace77f94be66aaeb2e Mon Sep 17 00:00:00 2001 From: Ning Xie Date: Sat, 1 Aug 2026 23:37:29 +0800 Subject: [PATCH] [model registry] some simple typos (#50673) Signed-off-by: Andy Xie --- vllm/model_executor/model_loader/utils.py | 6 +-- vllm/model_executor/models/registry.py | 46 +++++++++++------------ 2 files changed, 26 insertions(+), 26 deletions(-) diff --git a/vllm/model_executor/model_loader/utils.py b/vllm/model_executor/model_loader/utils.py index ecdcc16e9f..36683225d1 100644 --- a/vllm/model_executor/model_loader/utils.py +++ b/vllm/model_executor/model_loader/utils.py @@ -250,9 +250,9 @@ def get_model_architecture(model_config: ModelConfig) -> tuple[type[nn.Module], if key in _MODEL_ARCH_BY_HASH: return _MODEL_ARCH_BY_HASH[key] - model_arch = _get_model_architecture(model_config) - _MODEL_ARCH_BY_HASH[key] = model_arch - return model_arch + model_cls_and_arch = _get_model_architecture(model_config) + _MODEL_ARCH_BY_HASH[key] = model_cls_and_arch + return model_cls_and_arch def get_model_cls(model_config: ModelConfig) -> type[nn.Module]: diff --git a/vllm/model_executor/models/registry.py b/vllm/model_executor/models/registry.py index db3464a446..3798d22b6c 100644 --- a/vllm/model_executor/models/registry.py +++ b/vllm/model_executor/models/registry.py @@ -1094,7 +1094,7 @@ class _ModelRegistry: else: msg = ( "`model_cls` should be a string or PyTorch model class, " - f"not a {type(model_arch)}" + f"not a {type(model_cls)}" ) raise TypeError(msg) @@ -1352,88 +1352,88 @@ class _ModelRegistry: architectures: str | list[str], model_config: ModelConfig, ) -> bool: - model_cls, _ = self.inspect_model_cls(architectures, model_config) - return model_cls.is_text_generation_model + model_info, _ = self.inspect_model_cls(architectures, model_config) + return model_info.is_text_generation_model def is_pooling_model( self, architectures: str | list[str], model_config: ModelConfig, ) -> bool: - model_cls, _ = self.inspect_model_cls(architectures, model_config) - return model_cls.is_pooling_model + model_info, _ = self.inspect_model_cls(architectures, model_config) + return model_info.is_pooling_model def is_multimodal_model( self, architectures: str | list[str], model_config: ModelConfig, ) -> bool: - model_cls, _ = self.inspect_model_cls(architectures, model_config) - return model_cls.supports_multimodal + model_info, _ = self.inspect_model_cls(architectures, model_config) + return model_info.supports_multimodal def is_multimodal_raw_input_only_model( self, architectures: str | list[str], model_config: ModelConfig, ) -> bool: - model_cls, _ = self.inspect_model_cls(architectures, model_config) - return model_cls.supports_multimodal_raw_input_only + model_info, _ = self.inspect_model_cls(architectures, model_config) + return model_info.supports_multimodal_raw_input_only def is_pp_supported_model( self, architectures: str | list[str], model_config: ModelConfig, ) -> bool: - model_cls, _ = self.inspect_model_cls(architectures, model_config) - return model_cls.supports_pp + model_info, _ = self.inspect_model_cls(architectures, model_config) + return model_info.supports_pp def model_has_inner_state( self, architectures: str | list[str], model_config: ModelConfig, ) -> bool: - model_cls, _ = self.inspect_model_cls(architectures, model_config) - return model_cls.has_inner_state + model_info, _ = self.inspect_model_cls(architectures, model_config) + return model_info.has_inner_state def is_attention_free_model( self, architectures: str | list[str], model_config: ModelConfig, ) -> bool: - model_cls, _ = self.inspect_model_cls(architectures, model_config) - return model_cls.is_attention_free + model_info, _ = self.inspect_model_cls(architectures, model_config) + return model_info.is_attention_free def is_hybrid_model( self, architectures: str | list[str], model_config: ModelConfig, ) -> bool: - model_cls, _ = self.inspect_model_cls(architectures, model_config) - return model_cls.is_hybrid + model_info, _ = self.inspect_model_cls(architectures, model_config) + return model_info.is_hybrid def is_noops_model( self, architectures: str | list[str], model_config: ModelConfig, ) -> bool: - model_cls, _ = self.inspect_model_cls(architectures, model_config) - return model_cls.has_noops + model_info, _ = self.inspect_model_cls(architectures, model_config) + return model_info.has_noops def is_transcription_model( self, architectures: str | list[str], model_config: ModelConfig, ) -> bool: - model_cls, _ = self.inspect_model_cls(architectures, model_config) - return model_cls.supports_transcription + model_info, _ = self.inspect_model_cls(architectures, model_config) + return model_info.supports_transcription def is_transcription_only_model( self, architectures: str | list[str], model_config: ModelConfig, ) -> bool: - model_cls, _ = self.inspect_model_cls(architectures, model_config) - return model_cls.supports_transcription_only + model_info, _ = self.inspect_model_cls(architectures, model_config) + return model_info.supports_transcription_only def _resolve_module_name(mod_relname: str) -> str: