Skip to content

omnirt.core.registry

The model registry: the @register_model decorator, ModelSpec, ModelCapabilities, and discovery APIs. Auto-rendered reference: → API Reference / omnirt.core.registry.

Summary:

  • @register_model(id, *, task, capabilities=...) — decorator applied to a BasePipeline subclass to expose it in the registry.
  • ModelSpec — entry in the registry (id, task, pipeline class, capabilities).
  • ModelCapabilities — per-model metadata: tasks, adapters, resource_hint.min_vram_gb, recommended_preset, alias_of, summary.
  • register_model / get_model / list_models — programmatic API; the same data is what powers omnirt models and omnirt.list_available_models.

For how to onboard a new model, see Model Onboarding.