| from inference.core.managers.base import ModelManager | |
| class StubLoaderManager(ModelManager): | |
| def add_model(self, model_id: str, api_key: str) -> 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 = self.model_registry.get_model(model_id, api_key)( | |
| model_id=model_id, api_key=api_key, load_weights=False | |
| ) | |
| self._models[model_id] = model | |