From e8600013375de17cd59c2eb50bb3c972bbd7ef4e Mon Sep 17 00:00:00 2001 From: David Holtz Date: Wed, 14 Oct 2026 12:08:41 -0400 Subject: [PATCH] fix: reapply changes on latest --- src/diffusers/models/attention_dispatch.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/src/diffusers/models/attention_dispatch.py b/src/diffusers/models/attention_dispatch.py index 18e575d8ebbc..b1b91efbb58b 100644 --- a/src/diffusers/models/attention_dispatch.py +++ b/src/diffusers/models/attention_dispatch.py @@ -342,16 +342,16 @@ class _HubKernelConfig: AttentionBackendName.FLASH_HUB: _HubKernelConfig( repo_id="kernels-community/flash-attn2", function_attr="flash_attn_func", - wrapped_forward_attr="flash_attn_interface._flash_attn_forward", - wrapped_backward_attr="flash_attn_interface._flash_attn_backward", - version=3, + wrapped_forward_attr="flash_attn_interface._wrapped_flash_attn_forward", + wrapped_backward_attr="flash_attn_interface._wrapped_flash_attn_backward", + version=4, ), AttentionBackendName.FLASH_VARLEN_HUB: _HubKernelConfig( repo_id="kernels-community/flash-attn2", function_attr="flash_attn_varlen_func", - wrapped_forward_attr="flash_attn_interface._flash_attn_varlen_forward", - wrapped_backward_attr="flash_attn_interface._flash_attn_varlen_backward", - version=3, + wrapped_forward_attr="flash_attn_interface._wrapped_flash_attn_varlen_forward", + wrapped_backward_attr="flash_attn_interface._wrapped_flash_attn_varlen_backward", + version=4, ), AttentionBackendName.SAGE_HUB: _HubKernelConfig( repo_id="SageAttention/sage-attention",