Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
renzhc
diffusers_dcu
Commits
0eb715d7
Unverified
Commit
0eb715d7
authored
Sep 14, 2023
by
Younes Belkada
Committed by
GitHub
Sep 14, 2023
Browse files
[`LoRA`] Make optional arguments explicit (#5038)
make optional arguments explciit
parent
3aa64128
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
1 addition
and
1 deletion
+1
-1
src/diffusers/models/transformer_2d.py
src/diffusers/models/transformer_2d.py
+1
-1
No files found.
src/diffusers/models/transformer_2d.py
View file @
0eb715d7
...
@@ -284,7 +284,7 @@ class Transformer2DModel(ModelMixin, ConfigMixin):
...
@@ -284,7 +284,7 @@ class Transformer2DModel(ModelMixin, ConfigMixin):
hidden_states
=
self
.
norm
(
hidden_states
)
hidden_states
=
self
.
norm
(
hidden_states
)
if
not
self
.
use_linear_projection
:
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
]
inner_dim
=
hidden_states
.
shape
[
1
]
hidden_states
=
hidden_states
.
permute
(
0
,
2
,
3
,
1
).
reshape
(
batch
,
height
*
width
,
inner_dim
)
hidden_states
=
hidden_states
.
permute
(
0
,
2
,
3
,
1
).
reshape
(
batch
,
height
*
width
,
inner_dim
)
else
:
else
:
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment