Skip to content

Commit 1abc47f

Browse files
committed
tmp: Increase token count for transfer
Signed-off-by: Jacky <[email protected]>
1 parent e709de8 commit 1abc47f

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/fault_tolerance/cancellation/test_trtllm.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -82,7 +82,7 @@ def __init__(
8282
if mode != "prefill_and_decode":
8383
with open("test_request_cancellation_trtllm_config.yaml", "w") as f:
8484
f.write(
85-
"cache_transceiver_config:\n backend: DEFAULT\n max_tokens_in_buffer: 16384\n"
85+
"cache_transceiver_config:\n backend: DEFAULT\n max_tokens_in_buffer: 32767\n"
8686
)
8787
f.write("disable_overlap_scheduler: true\n")
8888
f.write("kv_cache_config:\n max_tokens: 16384\n")

0 commit comments

Comments
 (0)