Update benchmark_vllm_throughput.py (#12414)

This commit is contained in:
Guancheng Fu 2024-11-19 10:41:43 +08:00 committed by GitHub
parent a69395f31f
commit d6057f6dd2
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -388,4 +388,5 @@ if __name__ == "__main__":
if args.tokenizer != args.model: if args.tokenizer != args.model:
raise ValueError("Tokenizer must be the same as the model for MII " raise ValueError("Tokenizer must be the same as the model for MII "
"backend.") "backend.")
main(args)