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
chenpangpang
ComfyUI
Commits
537f35c7
"git@developer.sourcefind.cn:chenpangpang/ComfyUI.git" did not exist on "6fb084f39de495ca2cdfd482720069378d6e68ae"
Commit
537f35c7
authored
Jul 02, 2024
by
comfyanonymous
Browse files
Don't update dict if contiguous.
parent
3f46362d
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
4 additions
and
1 deletion
+4
-1
comfy/sd.py
comfy/sd.py
+4
-1
No files found.
comfy/sd.py
View file @
537f35c7
...
@@ -627,7 +627,10 @@ def save_checkpoint(output_path, model, clip=None, vae=None, clip_vision=None, m
...
@@ -627,7 +627,10 @@ def save_checkpoint(output_path, model, clip=None, vae=None, clip_vision=None, m
sd
=
model
.
model
.
state_dict_for_saving
(
clip_sd
,
vae
.
get_sd
(),
clip_vision_sd
)
sd
=
model
.
model
.
state_dict_for_saving
(
clip_sd
,
vae
.
get_sd
(),
clip_vision_sd
)
for
k
in
extra_keys
:
for
k
in
extra_keys
:
sd
[
k
]
=
extra_keys
[
k
]
sd
[
k
]
=
extra_keys
[
k
]
for
k
in
sd
:
for
k
in
sd
:
sd
[
k
]
=
sd
[
k
].
contiguous
()
t
=
sd
[
k
]
if
not
t
.
is_contiguous
():
sd
[
k
]
=
t
.
contiguous
()
comfy
.
utils
.
save_torch_file
(
sd
,
output_path
,
metadata
=
metadata
)
comfy
.
utils
.
save_torch_file
(
sd
,
output_path
,
metadata
=
metadata
)
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