Fix missing frames in SaveWEBM node.

2025-03-15 05:57:20 +00:00 · 2025-02-25 20:21:03 -05:00 · 2025-02-25 20:21:03 -05:00 · 0c32f82298
commit 0c32f82298
parent 189da3726d
1 changed files with 1 additions and 0 deletions
--- a/comfy_extras/nodes_video.py
+++ b/comfy_extras/nodes_video.py
@ -59,6 +59,7 @@ class SaveWEBM:
            frame = av.VideoFrame.from_ndarray(torch.clamp(frame[..., :3] * 255, min=0, max=255).to(device=torch.device("cpu"), dtype=torch.uint8).numpy(), format="rgb24")
            for packet in stream.encode(frame):
                container.mux(packet)
+        container.mux(stream.encode())
        container.close()

        results = [{