Complete ControlNet Support

2024-03-22 15:31:18 -07:00
parent ccd9611e2b
commit f7ad172ae3
1 changed files with 1 additions and 0 deletions
@@ -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]]),