Skip to content

GitLab

"...docs/git@developer.sourcefind.cn:wangsen/paddle_dbnet.git" did not exist on "8d99bef1c9f892f9d0035f040be6dfc62f2af9ae"
Commit 32be3582 authored by comfyanonymous's avatar comfyanonymous
Browse files

Save SD3 modelspec.architecture in CheckpointSave node.

parent 694e0b48
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
comfy_extras/nodes_model_merging.py
View file @ 32be3582
...@@ -183,6 +183,8 @@ def save_checkpoint(model, clip=None, vae=None, clip_vision=None, filename_prefi ...@@ -183,6 +183,8 @@ def save_checkpoint(model, clip=None, vae=None, clip_vision=None, filename_prefi
metadata["modelspec.architecture"] = "stable-diffusion-xl-v1-refiner" metadata["modelspec.architecture"] = "stable-diffusion-xl-v1-refiner"
elif isinstance(model.model, comfy.model_base.SVD_img2vid): elif isinstance(model.model, comfy.model_base.SVD_img2vid):
metadata["modelspec.architecture"] = "stable-video-diffusion-img2vid-v1" metadata["modelspec.architecture"] = "stable-video-diffusion-img2vid-v1"
elif isinstance(model.model, comfy.model_base.SD3):
metadata["modelspec.architecture"] = "stable-diffusion-v3-medium" #TODO: other SD3 variants
else: else:
enable_modelspec = False enable_modelspec = False
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment