Fix break caused by change to ComfyUI

2024-03-03 16:36:46 +02:00
parent a02e383454
commit 7fbd553c66
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