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
0850b88f
Unverified
Commit
0850b88f
authored
Feb 13, 2023
by
Plat
Committed by
GitHub
Feb 13, 2023
Browse files
Fix typo in load_pipeline_from_original_stable_diffusion_ckpt() method (#2320)
fix typo
parent
5d4f59ee
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
1 addition
and
1 deletion
+1
-1
src/diffusers/pipelines/stable_diffusion/convert_from_ckpt.py
...diffusers/pipelines/stable_diffusion/convert_from_ckpt.py
+1
-1
No files found.
src/diffusers/pipelines/stable_diffusion/convert_from_ckpt.py
View file @
0850b88f
...
...
@@ -871,7 +871,7 @@ def load_pipeline_from_original_stable_diffusion_ckpt(
if
original_config_file
is
None
:
key_name
=
"model.diffusion_model.input_blocks.2.1.transformer_blocks.0.attn2.to_k.weight"
original_config_file
=
os
.
path
.
join
(
tmpdir
,
"inferenc.yaml"
)
original_config_file
=
os
.
path
.
join
(
tmpdir
,
"inferenc
e
.yaml"
)
if
key_name
in
checkpoint
and
checkpoint
[
key_name
].
shape
[
-
1
]
==
1024
:
if
not
os
.
path
.
isfile
(
"v2-inference-v.yaml"
):
# model_type = "v2"
...
...
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