Skip to content

Commit 33becab

Browse files
Update repo_id for FLASH_4_HUB in attention_dispatch (#13822)
1 parent d94ad81 commit 33becab

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

src/diffusers/models/attention_dispatch.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -362,7 +362,7 @@ class _HubKernelConfig:
362362
version=1,
363363
),
364364
AttentionBackendName.FLASH_4_HUB: _HubKernelConfig(
365-
repo_id="kernels-staging/flash-attn4",
365+
repo_id="kernels-community/flash-attn4",
366366
function_attr="flash_attn_func",
367367
version=0,
368368
),

0 commit comments

Comments
 (0)