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
d67c3051
Commit
d67c3051
authored
Jan 03, 2023
by
Patrick von Platen
Browse files
allow conversion from no state dict checkpoints
parent
2bd53a94
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
3 additions
and
1 deletion
+3
-1
scripts/convert_original_stable_diffusion_to_diffusers.py
scripts/convert_original_stable_diffusion_to_diffusers.py
+3
-1
No files found.
scripts/convert_original_stable_diffusion_to_diffusers.py
View file @
d67c3051
...
@@ -866,6 +866,8 @@ if __name__ == "__main__":
...
@@ -866,6 +866,8 @@ if __name__ == "__main__":
else
:
else
:
print
(
"global_step key not found in model"
)
print
(
"global_step key not found in model"
)
global_step
=
None
global_step
=
None
if
"state_dict"
in
checkpoint
:
checkpoint
=
checkpoint
[
"state_dict"
]
checkpoint
=
checkpoint
[
"state_dict"
]
upcast_attention
=
False
upcast_attention
=
False
...
...
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