"ssh:/git@developer.sourcefind.cn:2222/OpenDAS/vllm_cscc.git" did not exist on "6791e80c77c1b9d1593f057a0c73c8a4c65db12a"
Unverified Commit c52ec5f0 authored by Kuntai Du's avatar Kuntai Du Committed by GitHub
Browse files

[Bugfix] fixing sonnet benchmark bug in benchmark_serving.py (#8616)

parent 02c9afa2
...@@ -626,9 +626,9 @@ def main(args: argparse.Namespace): ...@@ -626,9 +626,9 @@ def main(args: argparse.Namespace):
prefix_len=args.sonnet_prefix_len, prefix_len=args.sonnet_prefix_len,
tokenizer=tokenizer, tokenizer=tokenizer,
) )
input_requests = [(prompt, prompt_len, output_len) input_requests = [(prompt, prompt_len, output_len, None)
for prompt, prompt_formatted, prompt_len, for prompt, prompt_formatted, prompt_len,
output_len in input_requests] output_len, _ in input_requests]
else: else:
assert ( assert (
tokenizer.chat_template or tokenizer.default_chat_template tokenizer.chat_template or tokenizer.default_chat_template
...@@ -641,9 +641,9 @@ def main(args: argparse.Namespace): ...@@ -641,9 +641,9 @@ def main(args: argparse.Namespace):
prefix_len=args.sonnet_prefix_len, prefix_len=args.sonnet_prefix_len,
tokenizer=tokenizer, tokenizer=tokenizer,
) )
input_requests = [(prompt_formatted, prompt_len, output_len) input_requests = [(prompt_formatted, prompt_len, output_len, None)
for prompt, prompt_formatted, prompt_len, for prompt, prompt_formatted, prompt_len,
output_len in input_requests] output_len, _ in input_requests]
elif args.dataset_name == "hf": elif args.dataset_name == "hf":
input_requests = sample_hf_requests( input_requests = sample_hf_requests(
...@@ -963,4 +963,4 @@ if __name__ == "__main__": ...@@ -963,4 +963,4 @@ if __name__ == "__main__":
) )
args = parser.parse_args() args = parser.parse_args()
main(args) main(args)
\ No newline at end of file
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