1fee098f10
Signed-off-by: <> Co-authored-by: William Lin <SolitaryThinker@users.noreply.github.com> Co-authored-by: Will Lin <wlsaidhi@gmail.com> Co-authored-by: Zhou, Wei <wzhou322@gatech.edu> Co-authored-by: Kevin Lin <42618777+kevin314@users.noreply.github.com> Co-authored-by: kevin314 <kevin.lin.cs1@gmail.com> Co-authored-by: JerryZhou54 <69577934+JerryZhou54@users.noreply.github.com> Co-authored-by: Yongqi Chen <144848849+BrianChen1129@users.noreply.github.com> Co-authored-by: Peiyuan Zhang <m2deng@ucsd.edu>
40 lines
1.2 KiB
Python
40 lines
1.2 KiB
Python
# SPDX-License-Identifier: Apache-2.0
|
|
# adapted from vllm: https://github.com/vllm-project/vllm/blob/v0.7.3/vllm/entrypoints/cli/main.py
|
|
|
|
from typing import List
|
|
|
|
from fastvideo.v1.entrypoints.cli.cli_types import CLISubcommand
|
|
from fastvideo.v1.entrypoints.cli.generate import cmd_init as generate_cmd_init
|
|
from fastvideo.v1.utils import FlexibleArgumentParser
|
|
|
|
|
|
def cmd_init() -> List[CLISubcommand]:
|
|
"""Initialize all commands from separate modules"""
|
|
commands = []
|
|
commands.extend(generate_cmd_init())
|
|
return commands
|
|
|
|
|
|
def main() -> None:
|
|
parser = FlexibleArgumentParser(description="FastVideo CLI")
|
|
parser.add_argument('-v', '--version', action='version', version='0.1.0')
|
|
|
|
subparsers = parser.add_subparsers(required=False, dest="subparser")
|
|
|
|
cmds = {}
|
|
for cmd in cmd_init():
|
|
cmd.subparser_init(subparsers).set_defaults(dispatch_function=cmd.cmd)
|
|
cmds[cmd.name] = cmd
|
|
args = parser.parse_args()
|
|
if args.subparser in cmds:
|
|
cmds[args.subparser].validate(args)
|
|
|
|
if hasattr(args, "dispatch_function"):
|
|
args.dispatch_function(args)
|
|
else:
|
|
parser.print_help()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|