From 22f36de34e76b9f9ca79ac1323dd750ba78b79a6 Mon Sep 17 00:00:00 2001 From: Gaohan123 Date: Fri, 24 Oct 2025 14:09:56 +0800 Subject: [PATCH] fix error --- vllm_omni/model_executor/models/registry.py | 54 --------------------- 1 file changed, 54 deletions(-) delete mode 100644 vllm_omni/model_executor/models/registry.py diff --git a/vllm_omni/model_executor/models/registry.py b/vllm_omni/model_executor/models/registry.py deleted file mode 100644 index b01e17fa0..000000000 --- a/vllm_omni/model_executor/models/registry.py +++ /dev/null @@ -1,54 +0,0 @@ -from vllm.model_executor.models.registry import ( - _VLLM_MODELS, - _LazyRegisteredModel, - _ModelRegistry, -) - -_OMNI_MODELS = { - "Qwen2_5OmniForConditionalGeneration": ( - "qwen2_5_omni", - "Qwen2_5OmniForConditionalGeneration", - ), - "Qwen2_5OmniThinkerModel": ( - "qwen2_5_omni_thinker", - "Qwen2_5OmniThinkerForConditionalGeneration", - ), # noqa: E501 - "Qwen2_5OmniTalkerModel": ( - "qwen2_5_omni_talker", - "Qwen2_5OmniTalkerForConditionalGeneration", - ), # noqa: E501 - "Qwen2_5OmniToken2WavModel": ( - "qwen2_5_omni_token2wav", - "Qwen2_5OmniToken2WavForConditionalGenerationVLLM", - ), - "Qwen2_5OmniToken2WavDiTModel": ( - "qwen2_5_omni_token2wav", - "Qwen2_5OmniToken2WavModel", - ), - "Qwen2ForCausalLM_old": ("qwen2_old", "Qwen2ForCausalLM"), # need to discuss -} - -_VLLM_OMNI_MODELS = { - **_VLLM_MODELS, - **_OMNI_MODELS, -} - - -OmniModelRegistry = _ModelRegistry( - { - **{ - model_arch: _LazyRegisteredModel( - module_name=f"vllm.model_executor.models.{mod_relname}", - class_name=cls_name, - ) - for model_arch, (mod_relname, cls_name) in _VLLM_MODELS.items() - }, - **{ - model_arch: _LazyRegisteredModel( - module_name=f"vllm_omni.model_executor.models.{mod_relname}", - class_name=cls_name, - ) - for model_arch, (mod_relname, cls_name) in _OMNI_MODELS.items() - }, - } -)