vllm_omni.model_executor.models.registry ¶

OmniModelRegistry = _ModelRegistry(
    {
        **{
            model_arch: _LazyRegisteredModel(
                module_name=_resolve_module_name(
                    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_folder}.{mod_relname}",
                class_name=cls_name,
            )
            for model_arch, (
                mod_folder,
                mod_relname,
                cls_name,
            ) in _OMNI_MODELS.items()
        },
    }
)