From f7ad172ae3fcc5dc06342d9b50d195c17cc90b08 Mon Sep 17 00:00:00 2001
From: ExponentialML <59846140+ExponentialML@users.noreply.github.com>
Date: Fri, 22 Mar 2024 15:31:18 -0700
Subject: [PATCH] Complete ControlNet Support

---
 nodes.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/nodes.py b/nodes.py
index c0b239d..6055180 100644
--- a/nodes.py
+++ b/nodes.py
@@ -124,6 +124,7 @@ class DynamiCrafterProcessor:
         
         outs = []
         for i in range(batch_size):
+            model_in_kwargs['transformer_options']['cond_idx'] = i
             x_out = apply_model(
                     x_in[[i]], 
                     t=torch.cat([t[:1]]),