for the new diffuser version

pull/338/head
Hu Ye 2024-04-16 18:46:58 +08:00 committed by GitHub
parent 8e1a977274
commit 1fc70e162b
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 7 additions and 1 deletions

View File

@ -36,6 +36,8 @@ class LoRAAttnProcessor(nn.Module):
encoder_hidden_states=None,
attention_mask=None,
temb=None,
*args,
**kwargs,
):
residual = hidden_states
@ -130,6 +132,8 @@ class LoRAIPAttnProcessor(nn.Module):
encoder_hidden_states=None,
attention_mask=None,
temb=None,
*args,
**kwargs,
):
residual = hidden_states
@ -236,6 +240,8 @@ class LoRAAttnProcessor2_0(nn.Module):
encoder_hidden_states=None,
attention_mask=None,
temb=None,
*args,
**kwargs,
):
residual = hidden_states
@ -335,7 +341,7 @@ class LoRAIPAttnProcessor2_0(nn.Module):
self.to_v_ip = nn.Linear(cross_attention_dim or hidden_size, hidden_size, bias=False)
def __call__(
self, attn, hidden_states, encoder_hidden_states=None, attention_mask=None, scale=1.0, temb=None
self, attn, hidden_states, encoder_hidden_states=None, attention_mask=None, scale=1.0, temb=None, *args, **kwargs,
):
residual = hidden_states