Skip to content

GitLab

Unverified Commit f48c6791 authored by Zhuohan Li's avatar Zhuohan Li Committed by GitHub
Browse files

[FIX] Fix prefix test error on main (#3286)

parent c2c5e090
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 2 deletions
vllm/model_executor/layers/attention/backends/flash_attn.py
View file @ f48c6791
...@@ -103,8 +103,6 @@ class FlashAttentionBackend: ...@@ -103,8 +103,6 @@ class FlashAttentionBackend:
key_cache, key_cache,
value_cache, value_cache,
input_metadata, input_metadata,
self.num_heads,
self.num_kv_heads,
self.alibi_slopes, self.alibi_slopes,
) )
else: else:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment