From a9a0082d7cbcd1cbec74b79afac0462e9340c90c Mon Sep 17 00:00:00 2001
From: Jedrzej Kosinski <kosinkadink1@gmail.com>
Date: Fri, 22 Mar 2024 01:50:07 -0500
Subject: [PATCH] Added None check in calc_cond_uncond_batch_wrapper (uncond
 can be None)

---
 animatediff/sampling.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/animatediff/sampling.py b/animatediff/sampling.py
index 76ceae5..bf577b2 100644
--- a/animatediff/sampling.py
+++ b/animatediff/sampling.py
@@ -562,6 +562,8 @@ def calc_cond_uncond_batch_wrapper(model, cond, uncond, x_in, timestep, model_op
     # check if conds or unconds contain lora_hook
     contains_lora_hooks = False
     for cond_uncond in [cond, uncond]:
+        if cond_uncond is None:
+            continue
         for t in cond_uncond:
             if "lora_hook" in t:
                 contains_lora_hooks = True