[Minor] Fix benchmark_throughput.py (#151)

This commit is contained in:
Woosuk Kwon 2023-06-16 21:00:52 -07:00 committed by GitHub
parent eedb46bf03
commit bab8f3dd0d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -97,6 +97,7 @@ def run_cacheflow(
# FIXME(woosuk): Do not use internal method.
llm._add_request(
prompt=prompt,
prompt_token_ids=None,
sampling_params=sampling_params,
)