github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-09-14 21:51:25 +00:00
Code Issues Packages Projects Releases Wiki Activity

chore: fix code lint

Browse Source
This commit is contained in:
FangYin Cheng
2023-11-20 15:29:00 +08:00
parent d6318c21ee
commit 0fff10ba03
1 changed files with 1 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
pilot/utils/benchmarks/llm/llm_benchmarks.py
View File

@@ -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"