From 8949fdc440fa3d4e71610085e54dffda478c846b Mon Sep 17 00:00:00 2001 From: amy-why-3459 Date: Wed, 3 Jun 2026 23:44:46 +0800 Subject: [PATCH] [BugFix] Fix the issue of vllm failing to start. (#4105) Signed-off-by: amy-why-3459 --- vllm_omni/engine/arg_utils.py | 9 +++------ 1 file changed, 3 insertions(+), 6 deletions(-) diff --git a/vllm_omni/engine/arg_utils.py b/vllm_omni/engine/arg_utils.py index 3e5ce5536..25244c462 100644 --- a/vllm_omni/engine/arg_utils.py +++ b/vllm_omni/engine/arg_utils.py @@ -77,13 +77,10 @@ def register_omni_models_to_vllm(): _register_omni_hf_configs() - # Always register every arch declared in _OMNI_MODELS — these are the - # archs vllm-omni explicitly owns, so we want our implementation to - # win even if a future vllm release ships a same-named built-in. - # vllm's ModelRegistry.register_model already emits a warning and - # overrides on collision; we let it handle that. + supported_archs = ModelRegistry.get_supported_archs() for arch, (mod_folder, mod_relname, cls_name) in _OMNI_MODELS.items(): - ModelRegistry.register_model(arch, f"vllm_omni.model_executor.models.{mod_folder}.{mod_relname}:{cls_name}") + if arch not in supported_archs: + ModelRegistry.register_model(arch, f"vllm_omni.model_executor.models.{mod_folder}.{mod_relname}:{cls_name}") @dataclass