Merge branch 'main' into develop

2024-03-07 03:54:54 -06:00
parent 3acf378866 8afa5e0db7
commit 362e1b8bb3
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