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
2de3b69b
Commit
2de3b69b
authored
May 13, 2024
by
comfyanonymous
Browse files
Support saving some more modelspec types.
parent
cf6e1efb
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
6 additions
and
1 deletion
+6
-1
comfy_extras/nodes_model_merging.py
comfy_extras/nodes_model_merging.py
+6
-1
No files found.
comfy_extras/nodes_model_merging.py
View file @
2de3b69b
...
...
@@ -175,9 +175,14 @@ def save_checkpoint(model, clip=None, vae=None, clip_vision=None, filename_prefi
enable_modelspec
=
True
if
isinstance
(
model
.
model
,
comfy
.
model_base
.
SDXL
):
metadata
[
"modelspec.architecture"
]
=
"stable-diffusion-xl-v1-base"
if
isinstance
(
model
.
model
,
comfy
.
model_base
.
SDXL_instructpix2pix
):
metadata
[
"modelspec.architecture"
]
=
"stable-diffusion-xl-v1-edit"
else
:
metadata
[
"modelspec.architecture"
]
=
"stable-diffusion-xl-v1-base"
elif
isinstance
(
model
.
model
,
comfy
.
model_base
.
SDXLRefiner
):
metadata
[
"modelspec.architecture"
]
=
"stable-diffusion-xl-v1-refiner"
elif
isinstance
(
model
.
model
,
comfy
.
model_base
.
SVD_img2vid
):
metadata
[
"modelspec.architecture"
]
=
"stable-video-diffusion-img2vid-v1"
else
:
enable_modelspec
=
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