wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-06-17 05:20:18 +08:00
Code Issues Actions 19 Packages Projects Releases Wiki Activity

Add latent output to SaveLatent node

Browse Source
This commit is contained in:
Alexis Rolland 2026-05-13 13:46:40 +08:00
parent d05f18410a
commit 65465a2ea7
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
nodes.py
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: