23 lines
836 B
Python
23 lines
836 B
Python
from fastvideo.distributed import (maybe_init_distributed_environment_and_model_parallel)
|
|
from fastvideo.fastvideo_args import FastVideoArgs
|
|
from fastvideo.logger import init_logger
|
|
from fastvideo.utils import FlexibleArgumentParser
|
|
from fastvideo.workflow.workflow_base import WorkflowBase
|
|
|
|
logger = init_logger(__name__)
|
|
|
|
|
|
def main(fastvideo_args: FastVideoArgs) -> None:
|
|
maybe_init_distributed_environment_and_model_parallel(1, 1)
|
|
preprocess_workflow_cls = WorkflowBase.get_workflow_cls(fastvideo_args)
|
|
preprocess_workflow = preprocess_workflow_cls(fastvideo_args)
|
|
preprocess_workflow.run()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
parser = FlexibleArgumentParser()
|
|
parser = FastVideoArgs.add_cli_args(parser)
|
|
args = parser.parse_args()
|
|
fastvideo_args = FastVideoArgs.from_cli_args(args)
|
|
main(fastvideo_args)
|