diffengine.models.archs.ip_adapter

Module Contents

Functions

set_unet_ip_adapter(unet)

Set IP-Adapter for Unet.

load_ip_adapter(unet, image_projection, ...)

Load IP-Adapter pretrained weights.

process_ip_adapter_state_dict(unet, image_projection)

Process IP-Adapter state dict.
diffengine.models.archs.ip_adapter.set_unet_ip_adapter(unet)[source]

Set IP-Adapter for Unet.

Args:

unet (nn.Module): The unet to set IP-Adapter.

Parameters:

unet (torch.nn.Module) –

Return type:

None

diffengine.models.archs.ip_adapter.load_ip_adapter(unet, image_projection, pretrained_adapter, subfolder, weights_name)[source]

Load IP-Adapter pretrained weights.

Reference to diffusers/loaders/ip_adapter.py. and diffusers/loaders/unet.py.

Parameters:

  • unet (torch.nn.Module) –

  • image_projection (torch.nn.Module) –

  • pretrained_adapter (str) –

  • subfolder (str) –

  • weights_name (str) –

Return type:

None

diffengine.models.archs.ip_adapter.process_ip_adapter_state_dict(unet, image_projection)[source]

Process IP-Adapter state dict.

Parameters:

  • unet (torch.nn.Module) –

  • image_projection (torch.nn.Module) –

Return type:

dict