Skip to content

Commit 480fb35

Browse files
leisuzzJ石页a-r-r-o-w
authored
[Bugfix] typo fix in NPU FA (#12129)
[Bugfix] typo error in npu FA Co-authored-by: J石页 <[email protected]> Co-authored-by: Aryan <[email protected]>
1 parent 38740dd commit 480fb35

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/diffusers/models/attention_dispatch.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -944,7 +944,7 @@ def _native_npu_attention(
944944
pse=None,
945945
scale=1.0 / math.sqrt(query.shape[-1]) if scale is None else scale,
946946
pre_tockens=65536,
947-
next_tokens=65536,
947+
next_tockens=65536,
948948
keep_prob=1.0 - dropout_p,
949949
sync=False,
950950
inner_precise=0,

0 commit comments

Comments
 (0)