AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-29 07:22:12 +03:00
Code Activity

fix

Browse Source
This commit is contained in:
Aryan
2025-07-16 21:41:09 +02:00
parent f859fdf7ba
commit e76fc948b0
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
src/diffusers/models/attention_dispatch.py
View File

@@ -899,6 +899,7 @@ def _templated_context_parallel_attention(
@_AttentionBackendRegistry.register(
AttentionBackendName.FLASH,
constraints=[_check_device, _check_qkv_dtype_bf16_or_fp16, _check_shape],
supports_context_parallel=True,
)
def _flash_attention(
query: torch.Tensor,