| 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 | |