Skip to content

Commit 36c86d2

Browse files
authored
Merge branch 'main' into parallel-shards-loading
2 parents ad2dd62 + 480fb35 commit 36c86d2

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)