28 lines
903 B
Python
28 lines
903 B
Python
import os
|
|
|
|
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.preprocess.preprocess_workflow_t2v import (
|
|
PreprocessWorkflowT2V)
|
|
|
|
logger = init_logger(__name__)
|
|
|
|
|
|
def main(fastvideo_args: FastVideoArgs) -> None:
|
|
maybe_init_distributed_environment_and_model_parallel(1, 1)
|
|
num_gpus = int(os.environ["WORLD_SIZE"])
|
|
assert num_gpus == 1, "Only support 1 GPU"
|
|
preprocess_workflow = PreprocessWorkflowT2V(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)
|