Add latent output to SaveLatent node

This commit is contained in:
Alexis Rolland 2026-05-13 13:46:40 +08:00
parent d05f18410a
commit 65465a2ea7

View File

@ -488,7 +488,8 @@ class SaveLatent:
"filename_prefix": ("STRING", {"default": "latents/ComfyUI"})}, "filename_prefix": ("STRING", {"default": "latents/ComfyUI"})},
"hidden": {"prompt": "PROMPT", "extra_pnginfo": "EXTRA_PNGINFO"}, "hidden": {"prompt": "PROMPT", "extra_pnginfo": "EXTRA_PNGINFO"},
} }
RETURN_TYPES = () RETURN_TYPES = ("LATENT",)
RETURN_NAMES = ("samples",)
FUNCTION = "save" FUNCTION = "save"
OUTPUT_NODE = True OUTPUT_NODE = True
@ -526,7 +527,7 @@ class SaveLatent:
output["latent_format_version_0"] = torch.tensor([]) output["latent_format_version_0"] = torch.tensor([])
comfy.utils.save_torch_file(output, file, metadata=metadata) comfy.utils.save_torch_file(output, file, metadata=metadata)
return { "ui": { "latents": results } } return { "ui": { "latents": results }, "result": (samples,) }
class LoadLatent: class LoadLatent: