Skip to content

Commit

Permalink
[LoRA] Make optional arguments explicit (#5038)
Browse files Browse the repository at this point in the history
make optional arguments explciit
  • Loading branch information
younesbelkada authored Sep 14, 2023
1 parent 3aa6412 commit 0eb715d
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/diffusers/models/transformer_2d.py
Original file line number Diff line number Diff line change
Expand Up @@ -284,7 +284,7 @@ def forward(

hidden_states = self.norm(hidden_states)
if not self.use_linear_projection:
hidden_states = self.proj_in(hidden_states, lora_scale)
hidden_states = self.proj_in(hidden_states, scale=lora_scale)
inner_dim = hidden_states.shape[1]
hidden_states = hidden_states.permute(0, 2, 3, 1).reshape(batch, height * width, inner_dim)
else:
Expand Down

0 comments on commit 0eb715d

Please sign in to comment.