From 3304ea72756863483d158167ef023894873103f7 Mon Sep 17 00:00:00 2001 From: Xibo Sun Date: Thu, 26 Sep 2024 14:12:26 +0800 Subject: [PATCH] fix the example scripts in parallel_inference_xdit.py --- .../parallel_inference_xdit.py | 16 ++++++++++------ 1 file changed, 10 insertions(+), 6 deletions(-) diff --git a/tools/parallel_inference/parallel_inference_xdit.py b/tools/parallel_inference/parallel_inference_xdit.py index e10f385..e4caf33 100644 --- a/tools/parallel_inference/parallel_inference_xdit.py +++ b/tools/parallel_inference/parallel_inference_xdit.py @@ -7,16 +7,20 @@ https://github.com/xdit-project/xDiT/blob/main/examples/cogvideox_example.py By using this code, the inference process is parallelized on multiple GPUs, and thus speeded up. -You can also use the run.sh file in the same folder to automate running this -code for batch generation of videos. - Usage: 1. pip install xfuser -2. run the following command to generate video -torchrun --nproc_per_node=4 cogvideox_xdit.py --model \ - --ring_degree 2 --use_cfg_parallel --height 480 --width 720 --num_frames 9 \ +2. mkdir results +3. run the following command to generate video +torchrun --nproc_per_node=4 parallel_inference_xdit.py \ + --model --ulysses_degree 1 --ring_degree 2 \ + --use_cfg_parallel --height 480 --width 720 --num_frames 9 \ --prompt 'A small dog.' +You can also use the run.sh file in the same folder to automate running this +code for batch generation of videos, by running: + +sh ./run.sh + """ import time