From 54f340ce12ee2e0c7ac246a15cd9e3e8ef260db9 Mon Sep 17 00:00:00 2001
From: Yousef Rafat <81116377+yousef-rafat@users.noreply.github.com>
Date: Fri, 3 Jul 2026 18:18:43 +0300
Subject: [PATCH] for coderabbit

---
 comfy_extras/nodes_images.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/comfy_extras/nodes_images.py b/comfy_extras/nodes_images.py
index bd3e76adc..48829c166 100644
--- a/comfy_extras/nodes_images.py
+++ b/comfy_extras/nodes_images.py
@@ -1144,6 +1144,9 @@ def _encode_image(
     codec.pix_fmt = stream_fmt
     codec.time_base = Fraction(1, 1)
 
+    if num_channels == 1:
+        img_np = img_np.squeeze(-1)
+
     frame = av.VideoFrame.from_ndarray(img_np, format=frame_fmt)
     if frame_fmt != stream_fmt:
         frame = frame.reformat(format=stream_fmt)