From 0fff10ba03dd754d9a017a73c126a8db53aa330b Mon Sep 17 00:00:00 2001 From: FangYin Cheng Date: Mon, 20 Nov 2023 15:29:00 +0800 Subject: [PATCH] chore: fix code lint --- pilot/utils/benchmarks/llm/llm_benchmarks.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/pilot/utils/benchmarks/llm/llm_benchmarks.py b/pilot/utils/benchmarks/llm/llm_benchmarks.py index becb78b00..b70742670 100644 --- a/pilot/utils/benchmarks/llm/llm_benchmarks.py +++ b/pilot/utils/benchmarks/llm/llm_benchmarks.py @@ -206,9 +206,7 @@ if __name__ == "__main__": parser.add_argument("--model_type", type=str, default="huggingface") parser.add_argument("--result_csv_file", type=str, default=None) parser.add_argument("--input_lens", type=str, default="8,8,256,1024") - parser.add_argument( - "--output_lens", type=str, default="256,512,1024,1024" - ) + parser.add_argument("--output_lens", type=str, default="256,512,1024,1024") parser.add_argument("--parallel_nums", type=str, default="1,2,4,16,32") parser.add_argument( "--remote_model", type=bool, default=False, help="Connect to remote model"