Unverified Commit 0eb715d7 authored by Younes Belkada's avatar Younes Belkada Committed by GitHub
Browse files

[`LoRA`] Make optional arguments explicit (#5038)

make optional arguments explciit
parent 3aa64128
......@@ -284,7 +284,7 @@ class Transformer2DModel(ModelMixin, ConfigMixin):
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:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment