mirror of
https://github.com/invoke-ai/InvokeAI.git
synced 2026-02-11 22:15:19 -05:00
5a3195f757
- Replace AnyModelLoader with ModelLoaderRegistry - Fix type check errors in multiple files - Remove apparently unneeded `get_model_config_enum()` method from model manager - Remove last vestiges of old model manager - Updated tests and documentation resolve conflict with seamless.py
39 lines
1.2 KiB
Python
39 lines
1.2 KiB
Python
# Copyright (c) 2024, Lincoln D. Stein and the InvokeAI Development Team
|
|
"""Class for IP Adapter model loading in InvokeAI."""
|
|
|
|
from pathlib import Path
|
|
from typing import Optional
|
|
|
|
import torch
|
|
|
|
from invokeai.backend.ip_adapter.ip_adapter import build_ip_adapter
|
|
from invokeai.backend.model_manager import (
|
|
AnyModel,
|
|
BaseModelType,
|
|
ModelFormat,
|
|
ModelRepoVariant,
|
|
ModelType,
|
|
SubModelType,
|
|
)
|
|
from invokeai.backend.model_manager.load import ModelLoader, ModelLoaderRegistry
|
|
|
|
|
|
@ModelLoaderRegistry.register(base=BaseModelType.Any, type=ModelType.IPAdapter, format=ModelFormat.InvokeAI)
|
|
class IPAdapterInvokeAILoader(ModelLoader):
|
|
"""Class to load IP Adapter diffusers models."""
|
|
|
|
def _load_model(
|
|
self,
|
|
model_path: Path,
|
|
model_variant: Optional[ModelRepoVariant] = None,
|
|
submodel_type: Optional[SubModelType] = None,
|
|
) -> AnyModel:
|
|
if submodel_type is not None:
|
|
raise ValueError("There are no submodels in an IP-Adapter model.")
|
|
model = build_ip_adapter(
|
|
ip_adapter_ckpt_path=model_path / "ip_adapter.bin",
|
|
device=torch.device("cpu"),
|
|
dtype=self._torch_dtype,
|
|
)
|
|
return model
|