1
0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00

[Bugfix] typo fix in NPU FA (#12129)

[Bugfix] typo error in npu FA

Co-authored-by: J石页 <jiangshuo9@h-partners.com>
Co-authored-by: Aryan <aryan@huggingface.co>
This commit is contained in:
Leo Jiang
2025-08-12 10:42:19 -06:00
committed by GitHub
parent 38740ddbd8
commit 480fb357a3

View File

@@ -944,7 +944,7 @@ def _native_npu_attention(
pse=None,
scale=1.0 / math.sqrt(query.shape[-1]) if scale is None else scale,
pre_tockens=65536,
next_tokens=65536,
next_tockens=65536,
keep_prob=1.0 - dropout_p,
sync=False,
inner_precise=0,