diff --git a/src/diffusers/models/lora.py b/src/diffusers/models/lora.py index cc8e3e231e2bd..a777bb93e1c84 100644 --- a/src/diffusers/models/lora.py +++ b/src/diffusers/models/lora.py @@ -164,7 +164,10 @@ def forward(self, hidden_states, scale: float = 1.0): hidden_states, self.weight, self.bias, self.stride, self.padding, self.dilation, self.groups ) else: - return super().forward(hidden_states) + (scale * self.lora_layer(hidden_states)) + original_outputs = F.conv2d( + hidden_states, self.weight, self.bias, self.stride, self.padding, self.dilation, self.groups + ) + return original_outputs + (scale * self.lora_layer(hidden_states)) class LoRACompatibleLinear(nn.Linear):