Skip to content

vllm_omni.model_executor.models.registry

OmniModelRegistry module-attribute

OmniModelRegistry = _ModelRegistry(
    {
        None: {
            model_arch: (
                _LazyRegisteredModel(
                    module_name=_resolve_module_name(
                        mod_relname
                    ),
                    class_name=cls_name,
                )
            )
            for model_arch, (mod_relname, cls_name) in (
                items()
            )
        },
        None: {
            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 (items())
        },
    }
)