huggingface/text-generation-inference
5
0
Fork 0
mirror of https://github.com/huggingface/text-generation-inference.git synced 2025-04-19 13:52:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix state.plan call to use positional arguments

Browse Source
This commit is contained in:
Curtis Ruck 2025-04-11 10:09:24 -04:00
parent 9a8d0462e1
commit fd92054e1d
No known key found for this signature in database
1 changed files with 9 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
server/text_generation_server/layers/attention/flashinfer.py
View File

@ -84,16 +84,16 @@ def use_prefill_with_paged_kv_state(
token = prefill_with_paged_kv_state.set(state) token = prefill_with_paged_kv_state.set(state)
try: try:
state.plan( state.plan(
qo_indptr=cu_seqlens, cu_seqlens,
paged_kv_indptr=indptr, indptr,
paged_kv_indices=block_tables, block_tables,
paged_kv_last_page_len=last_page_len, last_page_len,
num_qo_heads=num_heads, num_heads,
num_kv_heads=num_kv_heads, num_kv_heads,
head_dim=head_size, head_size,
kv_data_type=kv_dtype, page_size,
q_data_type=q_dtype, q_data_type=q_dtype,
page_size=page_size, kv_data_type=kv_dtype,
) )
yield yield
finally: finally: