File tree Expand file tree Collapse file tree 1 file changed +10
-6
lines changed Expand file tree Collapse file tree 1 file changed +10
-6
lines changed Original file line number Diff line number Diff line change 77By using this code, the inference process is parallelized on multiple GPUs,
88and thus speeded up.
99
10- You can also use the run.sh file in the same folder to automate running this
11- code for batch generation of videos.
12-
1310Usage:
14111. pip install xfuser
15- 2. run the following command to generate video
16- torchrun --nproc_per_node=4 cogvideox_xdit.py --model <cogvideox-model-path> \
17- --ring_degree 2 --use_cfg_parallel --height 480 --width 720 --num_frames 9 \
12+ 2. mkdir results
13+ 3. run the following command to generate video
14+ torchrun --nproc_per_node=4 parallel_inference_xdit.py \
15+ --model <cogvideox-model-path> --ulysses_degree 1 --ring_degree 2 \
16+ --use_cfg_parallel --height 480 --width 720 --num_frames 9 \
1817 --prompt 'A small dog.'
1918
19+ You can also use the run.sh file in the same folder to automate running this
20+ code for batch generation of videos, by running:
21+
22+ sh ./run.sh
23+
2024"""
2125
2226import time
You can’t perform that action at this time.
0 commit comments