From 0c32f822987ef2a3eb8f87f84b6b6d1f0b2f3832 Mon Sep 17 00:00:00 2001
From: comfyanonymous <comfyanonymous@protonmail.com>
Date: Tue, 25 Feb 2025 20:21:03 -0500
Subject: [PATCH] Fix missing frames in SaveWEBM node.

---
 comfy_extras/nodes_video.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/comfy_extras/nodes_video.py b/comfy_extras/nodes_video.py
index f3922e03d90..53920ba1828 100644
--- 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 = [{