Revert "Fix break caused by change to ComfyUI"

2024-03-03 08:41:55 -06:00
parent 31ae334333
commit dc3f773bbf
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.get("image_only_indicator", None)
+        image_only_indicator = cond["image_only_indicator"]
        time_context = cond.get("time_context", None)
        del cond