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
2fa4b3ff
Unverified
Commit
2fa4b3ff
authored
Sep 04, 2023
by
Erwann Millon
Committed by
GitHub
Sep 04, 2023
Browse files
check for unet_lora_layers in sdxl pipeline's save_lora_weights method (#4821)
run make fix-copies and make style
parent
3201903d
Changes
5
Show whitespace changes
Inline
Side-by-side
Showing
5 changed files
with
35 additions
and
5 deletions
+35
-5
src/diffusers/pipelines/controlnet/pipeline_controlnet_inpaint_sd_xl.py
...pipelines/controlnet/pipeline_controlnet_inpaint_sd_xl.py
+7
-1
src/diffusers/pipelines/controlnet/pipeline_controlnet_sd_xl.py
...ffusers/pipelines/controlnet/pipeline_controlnet_sd_xl.py
+7
-1
src/diffusers/pipelines/stable_diffusion_xl/pipeline_stable_diffusion_xl.py
...lines/stable_diffusion_xl/pipeline_stable_diffusion_xl.py
+7
-1
src/diffusers/pipelines/stable_diffusion_xl/pipeline_stable_diffusion_xl_img2img.py
...able_diffusion_xl/pipeline_stable_diffusion_xl_img2img.py
+7
-1
src/diffusers/pipelines/stable_diffusion_xl/pipeline_stable_diffusion_xl_inpaint.py
...able_diffusion_xl/pipeline_stable_diffusion_xl_inpaint.py
+7
-1
No files found.
src/diffusers/pipelines/controlnet/pipeline_controlnet_inpaint_sd_xl.py
View file @
2fa4b3ff
...
@@ -1591,6 +1591,12 @@ class StableDiffusionXLControlNetInpaintPipeline(DiffusionPipeline, LoraLoaderMi
...
@@ -1591,6 +1591,12 @@ class StableDiffusionXLControlNetInpaintPipeline(DiffusionPipeline, LoraLoaderMi
layers_state_dict
=
{
f
"
{
prefix
}
.
{
module_name
}
"
:
param
for
module_name
,
param
in
layers_weights
.
items
()}
layers_state_dict
=
{
f
"
{
prefix
}
.
{
module_name
}
"
:
param
for
module_name
,
param
in
layers_weights
.
items
()}
return
layers_state_dict
return
layers_state_dict
if
not
(
unet_lora_layers
or
text_encoder_lora_layers
or
text_encoder_2_lora_layers
):
raise
ValueError
(
"You must pass at least one of `unet_lora_layers`, `text_encoder_lora_layers` or `text_encoder_2_lora_layers`."
)
if
unet_lora_layers
:
state_dict
.
update
(
pack_weights
(
unet_lora_layers
,
"unet"
))
state_dict
.
update
(
pack_weights
(
unet_lora_layers
,
"unet"
))
if
text_encoder_lora_layers
and
text_encoder_2_lora_layers
:
if
text_encoder_lora_layers
and
text_encoder_2_lora_layers
:
...
...
src/diffusers/pipelines/controlnet/pipeline_controlnet_sd_xl.py
View file @
2fa4b3ff
...
@@ -1254,6 +1254,12 @@ class StableDiffusionXLControlNetPipeline(
...
@@ -1254,6 +1254,12 @@ class StableDiffusionXLControlNetPipeline(
layers_state_dict
=
{
f
"
{
prefix
}
.
{
module_name
}
"
:
param
for
module_name
,
param
in
layers_weights
.
items
()}
layers_state_dict
=
{
f
"
{
prefix
}
.
{
module_name
}
"
:
param
for
module_name
,
param
in
layers_weights
.
items
()}
return
layers_state_dict
return
layers_state_dict
if
not
(
unet_lora_layers
or
text_encoder_lora_layers
or
text_encoder_2_lora_layers
):
raise
ValueError
(
"You must pass at least one of `unet_lora_layers`, `text_encoder_lora_layers` or `text_encoder_2_lora_layers`."
)
if
unet_lora_layers
:
state_dict
.
update
(
pack_weights
(
unet_lora_layers
,
"unet"
))
state_dict
.
update
(
pack_weights
(
unet_lora_layers
,
"unet"
))
if
text_encoder_lora_layers
and
text_encoder_2_lora_layers
:
if
text_encoder_lora_layers
and
text_encoder_2_lora_layers
:
...
...
src/diffusers/pipelines/stable_diffusion_xl/pipeline_stable_diffusion_xl.py
View file @
2fa4b3ff
...
@@ -953,6 +953,12 @@ class StableDiffusionXLPipeline(DiffusionPipeline, FromSingleFileMixin, LoraLoad
...
@@ -953,6 +953,12 @@ class StableDiffusionXLPipeline(DiffusionPipeline, FromSingleFileMixin, LoraLoad
layers_state_dict
=
{
f
"
{
prefix
}
.
{
module_name
}
"
:
param
for
module_name
,
param
in
layers_weights
.
items
()}
layers_state_dict
=
{
f
"
{
prefix
}
.
{
module_name
}
"
:
param
for
module_name
,
param
in
layers_weights
.
items
()}
return
layers_state_dict
return
layers_state_dict
if
not
(
unet_lora_layers
or
text_encoder_lora_layers
or
text_encoder_2_lora_layers
):
raise
ValueError
(
"You must pass at least one of `unet_lora_layers`, `text_encoder_lora_layers` or `text_encoder_2_lora_layers`."
)
if
unet_lora_layers
:
state_dict
.
update
(
pack_weights
(
unet_lora_layers
,
"unet"
))
state_dict
.
update
(
pack_weights
(
unet_lora_layers
,
"unet"
))
if
text_encoder_lora_layers
and
text_encoder_2_lora_layers
:
if
text_encoder_lora_layers
and
text_encoder_2_lora_layers
:
...
...
src/diffusers/pipelines/stable_diffusion_xl/pipeline_stable_diffusion_xl_img2img.py
View file @
2fa4b3ff
...
@@ -1112,6 +1112,12 @@ class StableDiffusionXLImg2ImgPipeline(DiffusionPipeline, FromSingleFileMixin, L
...
@@ -1112,6 +1112,12 @@ class StableDiffusionXLImg2ImgPipeline(DiffusionPipeline, FromSingleFileMixin, L
layers_state_dict
=
{
f
"
{
prefix
}
.
{
module_name
}
"
:
param
for
module_name
,
param
in
layers_weights
.
items
()}
layers_state_dict
=
{
f
"
{
prefix
}
.
{
module_name
}
"
:
param
for
module_name
,
param
in
layers_weights
.
items
()}
return
layers_state_dict
return
layers_state_dict
if
not
(
unet_lora_layers
or
text_encoder_lora_layers
or
text_encoder_2_lora_layers
):
raise
ValueError
(
"You must pass at least one of `unet_lora_layers`, `text_encoder_lora_layers` or `text_encoder_2_lora_layers`."
)
if
unet_lora_layers
:
state_dict
.
update
(
pack_weights
(
unet_lora_layers
,
"unet"
))
state_dict
.
update
(
pack_weights
(
unet_lora_layers
,
"unet"
))
if
text_encoder_lora_layers
and
text_encoder_2_lora_layers
:
if
text_encoder_lora_layers
and
text_encoder_2_lora_layers
:
...
...
src/diffusers/pipelines/stable_diffusion_xl/pipeline_stable_diffusion_xl_inpaint.py
View file @
2fa4b3ff
...
@@ -1426,6 +1426,12 @@ class StableDiffusionXLInpaintPipeline(DiffusionPipeline, LoraLoaderMixin, FromS
...
@@ -1426,6 +1426,12 @@ class StableDiffusionXLInpaintPipeline(DiffusionPipeline, LoraLoaderMixin, FromS
layers_state_dict
=
{
f
"
{
prefix
}
.
{
module_name
}
"
:
param
for
module_name
,
param
in
layers_weights
.
items
()}
layers_state_dict
=
{
f
"
{
prefix
}
.
{
module_name
}
"
:
param
for
module_name
,
param
in
layers_weights
.
items
()}
return
layers_state_dict
return
layers_state_dict
if
not
(
unet_lora_layers
or
text_encoder_lora_layers
or
text_encoder_2_lora_layers
):
raise
ValueError
(
"You must pass at least one of `unet_lora_layers`, `text_encoder_lora_layers` or `text_encoder_2_lora_layers`."
)
if
unet_lora_layers
:
state_dict
.
update
(
pack_weights
(
unet_lora_layers
,
"unet"
))
state_dict
.
update
(
pack_weights
(
unet_lora_layers
,
"unet"
))
if
text_encoder_lora_layers
and
text_encoder_2_lora_layers
:
if
text_encoder_lora_layers
and
text_encoder_2_lora_layers
:
...
...
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