Save SD3 modelspec.architecture in CheckpointSave node.

This commit is contained in:
comfyanonymous 2024-06-12 01:02:07 -04:00
parent 694e0b48e0
commit 32be358213
1 changed files with 2 additions and 0 deletions

View File

@ -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"
elif isinstance(model.model, comfy.model_base.SVD_img2vid):
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:
enable_modelspec = False