inference/core/managers/stub_loader.py

from inference.core.managers.base import ModelManager


class StubLoaderManager(ModelManager):
    def add_model(self, model_id: str, api_key: str, model_id_alias=None) -> None:
        """Adds a new model to the manager.

        Args:
            model_id (str): The identifier of the model.
            model (Model): The model instance.
        """
        if model_id in self._models:
            return
        model_class = self.model_registry.get_model(
            model_id_alias if model_id_alias is not None else model_id, api_key
        )
        model = model_class(model_id=model_id, api_key=api_key, load_weights=False)
        self._models[model_id] = model