mirror of
https://git.datalinker.icu/vllm-project/vllm.git
synced 2026-07-13 02:37:10 +08:00
[ROCM] fix native attention function call (#13650)
This commit is contained in:
parent
68d535ef44
commit
68d630a0c7
@ -717,7 +717,6 @@ class ROCmFlashAttentionImpl(AttentionImpl):
|
|||||||
self.num_heads,
|
self.num_heads,
|
||||||
self.head_size,
|
self.head_size,
|
||||||
self.scale,
|
self.scale,
|
||||||
causal_mask,
|
|
||||||
attn_masks,
|
attn_masks,
|
||||||
)
|
)
|
||||||
else:
|
else:
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user