Fix for latest upstream

This commit is contained in:
kijai 2026-01-05 00:24:16 +02:00
parent 7bc68588e8
commit c3e6609e20

View File

@ -436,7 +436,7 @@ class MultiTalkCrossAttnPatch:
def __call__(self, kwargs): def __call__(self, kwargs):
transformer_options = kwargs.get("transformer_options", {}) 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"] x = kwargs["x"]
if block_idx is None: if block_idx is None:
return torch.zeros_like(x) return torch.zeros_like(x)