Fix bug for new comfyui version

2026-02-13 21:49:19 +08:00
parent 38371ba52a
commit 0bebbdfb37
1 changed files with 10 additions and 7 deletions
@@ -60,14 +60,17 @@ class SaveVideoRGBA(io.ComfyNode):
                resized_images.append(resized_tensor.unsqueeze(0))

            images = torch.cat(resized_images, dim=0)
-
-        video = RGBAVideoFromComponents(
-            VideoComponents(
-                images=images,
-                audio=audio,
-                frame_rate=Fraction(fps),
+        try:
+            from comfy_api.latest import InputImpl, Types
+            video = InputImpl.VideoFromComponents(Types.VideoComponents(images=images, audio=audio, frame_rate=Fraction(fps)))
+        except Exception as e:
+            video = RGBAVideoFromComponents(
+                VideoComponents(
+                    images=images,
+                    audio=audio,
+                    frame_rate=Fraction(fps),
+                )
            )
-        )

        width, height = video.get_dimensions()
        results = list()