AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00
Code Activity

Make passing the IP Adapter mask to the attention mechanism optional (#10346)

Browse Source 
Make passing the IP Adapter mask to the attention mechanism optional if there is no need to apply it to a given IP Adapter.
This commit is contained in:
Eliseu Silva
2024-12-24 12:28:42 -03:00
committed by GitHub
parent 6dfaec3487
commit c0c11683f3
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/diffusers/models/attention_processor.py
View File

@@ -4839,6 +4839,8 @@ class IPAdapterAttnProcessor(nn.Module):
)
else:
for index, (mask, scale, ip_state) in enumerate(zip(ip_adapter_masks, self.scale, ip_hidden_states)):
if mask is None:
continue
if not isinstance(mask, torch.Tensor) or mask.ndim != 4:
raise ValueError(
"Each element of the ip_adapter_masks array should be a tensor with shape "
@@ -5056,6 +5058,8 @@ class IPAdapterAttnProcessor2_0(torch.nn.Module):
)
else:
for index, (mask, scale, ip_state) in enumerate(zip(ip_adapter_masks, self.scale, ip_hidden_states)):
if mask is None:
continue
if not isinstance(mask, torch.Tensor) or mask.ndim != 4:
raise ValueError(
"Each element of the ip_adapter_masks array should be a tensor with shape "