from .registry import OmniModelRegistry