Fix missing frames in SaveWEBM node.

This commit is contained in:
comfyanonymous 2025-02-25 20:21:03 -05:00
parent 189da3726d
commit 0c32f82298

View File

@ -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") 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): for packet in stream.encode(frame):
container.mux(packet) container.mux(packet)
container.mux(stream.encode())
container.close() container.close()
results = [{ results = [{