Skip to content

Commit

Permalink
Fix pre-commit hooks
Browse files Browse the repository at this point in the history
  • Loading branch information
nv-hwoo committed Nov 30, 2023
1 parent 0067a9d commit 6869902
Showing 1 changed file with 6 additions and 7 deletions.
13 changes: 6 additions & 7 deletions src/c++/perf_analyzer/docs/examples/profile.py
Original file line number Diff line number Diff line change
Expand Up @@ -421,12 +421,12 @@ def profile(args, export_file):
f"--profile-export-file={export_file} "
)
if args.model == "ensemble": # TRT-LLM
command += (
"--shape=text_input:1 "
"--shape=max_tokens:1 "
"--shape=bad_words:1 "
"--shape=stop_words:1 "
)
command += (
"--shape=text_input:1 "
"--shape=max_tokens:1 "
"--shape=bad_words:1 "
"--shape=stop_words:1 "
)
if args.periodic_concurrency_range:
start, end, step = args.periodic_concurrency_range
command += (
Expand Down Expand Up @@ -501,7 +501,6 @@ def construct_vllm_input_data(args):
}
]
}


# If command line option is specified, overwrite
if args.offline:
Expand Down

0 comments on commit 6869902

Please sign in to comment.