This commit is contained in:
Octopus 2026-04-30 16:32:11 -04:00 committed by GitHub
commit 1b2b4e4285
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 23 additions and 21 deletions

View File

@ -196,7 +196,7 @@ class SaveAnimatedWEBP(IO.ComfyNode):
IO.Float.Input("fps", default=6.0, min=0.01, max=1000.0, step=0.01), IO.Float.Input("fps", default=6.0, min=0.01, max=1000.0, step=0.01),
IO.Boolean.Input("lossless", default=True), IO.Boolean.Input("lossless", default=True),
IO.Int.Input("quality", default=80, min=0, max=100), IO.Int.Input("quality", default=80, min=0, max=100),
IO.Combo.Input("method", options=list(cls.COMPRESS_METHODS.keys())), IO.Combo.Input("method", options=list(cls.COMPRESS_METHODS.keys()), default="fastest"),
# "num_frames": ("INT", {"default": 0, "min": 0, "max": 8192}), # "num_frames": ("INT", {"default": 0, "min": 0, "max": 8192}),
], ],
hidden=[IO.Hidden.prompt, IO.Hidden.extra_pnginfo], hidden=[IO.Hidden.prompt, IO.Hidden.extra_pnginfo],

View File

@ -39,6 +39,7 @@ class SaveWEBM(io.ComfyNode):
file = f"{filename}_{counter:05}_.webm" file = f"{filename}_{counter:05}_.webm"
container = av.open(os.path.join(full_output_folder, file), mode="w") container = av.open(os.path.join(full_output_folder, file), mode="w")
try:
if cls.hidden.prompt is not None: if cls.hidden.prompt is not None:
container.metadata["prompt"] = json.dumps(cls.hidden.prompt) container.metadata["prompt"] = json.dumps(cls.hidden.prompt)
@ -61,6 +62,7 @@ class SaveWEBM(io.ComfyNode):
for packet in stream.encode(frame): for packet in stream.encode(frame):
container.mux(packet) container.mux(packet)
container.mux(stream.encode()) container.mux(stream.encode())
finally:
container.close() container.close()
return io.NodeOutput(ui=ui.PreviewVideo([ui.SavedResult(file, subfolder, io.FolderType.output)])) return io.NodeOutput(ui=ui.PreviewVideo([ui.SavedResult(file, subfolder, io.FolderType.output)]))