def METHOD_NAME(tag_like: str | Tag) -> Model:
    model = bentoml.models.METHOD_NAME(tag_like)
    if model.info.module not in (MODULE_NAME, __name__):
        raise NotFound(
            f"Model {model.tag} was saved with module {model.info.module}, not loading with {MODULE_NAME}."
        )
    return model