Merge PR #75 from harelc/fix_image_only_indicator_bug

Fix break caused by change to ComfyUI
2024-03-03 08:40:39 -06:00
parent 8be003562d 7fbd553c66
commit 31ae334333
1 changed files with 1 additions and 1 deletions
@@ -305,7 +305,7 @@ class SVDControlNet(nn.Module):

        cond = kwargs["cond"]
        num_video_frames = cond["num_video_frames"]
-        image_only_indicator = cond["image_only_indicator"]
+        image_only_indicator = cond.get("image_only_indicator", None)
        time_context = cond.get("time_context", None)
        del cond