Fix missing frames in SaveWEBM node.

comfyanonymous · Feb 26, 2025 · 0c32f82 · 0c32f82
1 parent 189da37
commit 0c32f82
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/comfy_extras/nodes_video.py b/comfy_extras/nodes_video.py
@@ -59,6 +59,7 @@ def save_images(self, images, codec, fps, filename_prefix, crf, prompt=None, ext
             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 = [{