[Misc] extract parser.parse_args() (#18323)

Signed-off-by: reidliu41 <reid201711@gmail.com>
Co-authored-by: reidliu41 <reid201711@gmail.com>
This commit is contained in:
Reid
2025-05-19 12:06:26 +08:00
committed by GitHub
parent 221cfc2fea
commit 27d0952600
2 changed files with 12 additions and 4 deletions

View File

@@ -140,7 +140,7 @@ def main(args):
print(generated_text)
if __name__ == "__main__":
def parse_args():
parser = FlexibleArgumentParser(
description='Demo on using vLLM for offline inference with '
'audio language models')
@@ -155,5 +155,9 @@ if __name__ == "__main__":
default=None,
help="Set the seed when initializing `vllm.LLM`.")
args = parser.parse_args()
return parser.parse_args()
if __name__ == "__main__":
args = parse_args()
main(args)