Author: hlky

src/diffusers/models/controlnets/controlnet_union.py

@@ -764,7 +764,7 @@ def forward(
             x = layer(x)
 
         controlnet_cond_fuser = sample * 0.0
-        for idx, condition in enumerate(condition_list[:-1]):
+        for idx, condition in zip(control_type_idx, condition_list[:-1]):
             alpha = self.spatial_ch_projs(x[:, idx])
             alpha = alpha.unsqueeze(-1).unsqueeze(-1)
             controlnet_cond_fuser += condition + alpha