diff --git a/comfy/ldm/wan/model_multitalk.py b/comfy/ldm/wan/model_multitalk.py index a67fb5158..c9dd98c4d 100644 --- a/comfy/ldm/wan/model_multitalk.py +++ b/comfy/ldm/wan/model_multitalk.py @@ -436,7 +436,7 @@ class MultiTalkCrossAttnPatch: def __call__(self, kwargs): transformer_options = kwargs.get("transformer_options", {}) - block_idx = transformer_options.get("block_idx", None) + block_idx = transformer_options.get("block_index", None) x = kwargs["x"] if block_idx is None: return torch.zeros_like(x)