Fabrice-TIERCELIN
commited on
Upload 2 files
Browse files
scripts/run_sample_video_fp8.sh
ADDED
@@ -0,0 +1,17 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
#!/bin/bash
|
2 |
+
# Description: This script demonstrates how to inference a video based on HunyuanVideo model
|
3 |
+
DIT_CKPT_PATH={PATH_TO}/{MODEL_NAME}_model_states_fp8.pt
|
4 |
+
|
5 |
+
python3 sample_video.py \
|
6 |
+
--dit-weight ${DIT_CKPT_PATH} \
|
7 |
+
--video-size 720 1280 \
|
8 |
+
--video-length 129 \
|
9 |
+
--infer-steps 50 \
|
10 |
+
--prompt "A cat walks on the grass, realistic style." \
|
11 |
+
--seed 42 \
|
12 |
+
--embedded-cfg-scale 6.0 \
|
13 |
+
--flow-shift 7.0 \
|
14 |
+
--flow-reverse \
|
15 |
+
--use-cpu-offload \
|
16 |
+
--use-fp8 \
|
17 |
+
--save-path ./results
|
scripts/run_sample_video_multigpu.sh
ADDED
@@ -0,0 +1,46 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
#!/bin/bash
|
2 |
+
# Description: This script demonstrates how to inference a video based on HunyuanVideo model
|
3 |
+
|
4 |
+
# Supported Parallel Configurations
|
5 |
+
# | --video-size | --video-length | --ulysses-degree x --ring-degree | --nproc_per_node |
|
6 |
+
# |----------------------|----------------|----------------------------------|------------------|
|
7 |
+
# | 1280 720 or 720 1280 | 129 | 8x1,4x2,2x4,1x8 | 8 |
|
8 |
+
# | 1280 720 or 720 1280 | 129 | 1x5 | 5 |
|
9 |
+
# | 1280 720 or 720 1280 | 129 | 4x1,2x2,1x4 | 4 |
|
10 |
+
# | 1280 720 or 720 1280 | 129 | 3x1,1x3 | 3 |
|
11 |
+
# | 1280 720 or 720 1280 | 129 | 2x1,1x2 | 2 |
|
12 |
+
# | 1104 832 or 832 1104 | 129 | 4x1,2x2,1x4 | 4 |
|
13 |
+
# | 1104 832 or 832 1104 | 129 | 3x1,1x3 | 3 |
|
14 |
+
# | 1104 832 or 832 1104 | 129 | 2x1,1x2 | 2 |
|
15 |
+
# | 960 960 | 129 | 6x1,3x2,2x3,1x6 | 6 |
|
16 |
+
# | 960 960 | 129 | 4x1,2x2,1x4 | 4 |
|
17 |
+
# | 960 960 | 129 | 3x1,1x3 | 3 |
|
18 |
+
# | 960 960 | 129 | 1x2,2x1 | 2 |
|
19 |
+
# | 960 544 or 544 960 | 129 | 6x1,3x2,2x3,1x6 | 6 |
|
20 |
+
# | 960 544 or 544 960 | 129 | 4x1,2x2,1x4 | 4 |
|
21 |
+
# | 960 544 or 544 960 | 129 | 3x1,1x3 | 3 |
|
22 |
+
# | 960 544 or 544 960 | 129 | 1x2,2x1 | 2 |
|
23 |
+
# | 832 624 or 624 832 | 129 | 4x1,2x2,1x4 | 4 |
|
24 |
+
# | 624 832 or 624 832 | 129 | 3x1,1x3 | 3 |
|
25 |
+
# | 832 624 or 624 832 | 129 | 2x1,1x2 | 2 |
|
26 |
+
# | 720 720 | 129 | 1x5 | 5 |
|
27 |
+
# | 720 720 | 129 | 3x1,1x3 | 3 |
|
28 |
+
|
29 |
+
export TOKENIZERS_PARALLELISM=false
|
30 |
+
|
31 |
+
export NPROC_PER_NODE=8
|
32 |
+
export ULYSSES_DEGREE=8
|
33 |
+
export RING_DEGREE=1
|
34 |
+
|
35 |
+
torchrun --nproc_per_node=$NPROC_PER_NODE sample_video.py \
|
36 |
+
--video-size 720 1280 \
|
37 |
+
--video-length 129 \
|
38 |
+
--infer-steps 50 \
|
39 |
+
--prompt "A cat walks on the grass, realistic style." \
|
40 |
+
--seed 42 \
|
41 |
+
--embedded-cfg-scale 6.0 \
|
42 |
+
--flow-shift 7.0 \
|
43 |
+
--flow-reverse \
|
44 |
+
--ulysses-degree=$ULYSSES_DEGREE \
|
45 |
+
--ring-degree=$RING_DEGREE \
|
46 |
+
--save-path ./results
|