From 7fbd553c666340eb5088f6b2216373f7b4d9b0f4 Mon Sep 17 00:00:00 2001
From: Harel Cain <harel@lightricks.com>
Date: Sun, 3 Mar 2024 13:37:26 +0200
Subject: [PATCH] Fix break caused by change to ComfyUI

---
 adv_control/control_svd.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/adv_control/control_svd.py b/adv_control/control_svd.py
index 6a7de04..9458f6a 100644
--- a/adv_control/control_svd.py
+++ b/adv_control/control_svd.py
@@ -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