nodes/hao-ai-lab-FastVideo
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[v0] Remove V0 code (#621)

Browse Source
This commit is contained in:
William Lin
2025-07-15 22:05:16 -07:00
committed by GitHub
parent b79d1fc15b
commit 576c214f28
119 changed files with 86 additions and 23605 deletions
Download Patch File Download Diff File Expand all files Collapse all files
scripts/inference/v0/inference_hunyuan.sh
-40
View File
@@ -1,40 +0,0 @@
#!/bin/bash
num_gpus=4
export MODEL_BASE=data/FastHunyuan
torchrun --nnodes=1 --nproc_per_node=$num_gpus --master_port 29503 \
fastvideo/sample/sample_t2v_hunyuan.py \
--height 720 \
--width 1280 \
--num_frames 125 \
--num_inference_steps 6 \
--guidance_scale 1 \
--embedded_cfg_scale 6 \
--flow_shift 17 \
--flow-reverse \
--prompt ./assets/prompt.txt \
--seed 1024 \
--output_path outputs_video/hunyuan/vae_sp/ \
--model_path $MODEL_BASE \
--dit-weight ${MODEL_BASE}/hunyuan-video-t2v-720p/transformers/mp_rank_00_model_states.pt \
--vae-sp
# Inference original Hunyuan
num_gpus=4
export MODEL_BASE=data/hunyuan
torchrun --nnodes=1 --nproc_per_node=$num_gpus --master_port 29503 \
fastvideo/sample/sample_t2v_hunyuan.py \
--height 720 \
--width 1280 \
--num_frames 125 \
--num_inference_steps 50 \
--guidance_scale 1 \
--embedded_cfg_scale 6 \
--flow_shift 7 \
--flow-reverse \
--prompt ./assets/prompt.txt \
--seed 1024 \
--output_path outputs_video/hunyuan/vae_sp/ \
--model_path $MODEL_BASE \
--dit-weight ${MODEL_BASE}/hunyuan-video-t2v-720p/transformers/mp_rank_00_model_states.pt \
--vae-sp
scripts/inference/v0/inference_hunyuan_STA.sh
-49
View File
@@ -1,49 +0,0 @@
#!/bin/bash
# Inference with STA + Teacache
num_gpus=1 # currently it should be a factor of 30
mask_strategy_file_path=assets/mask_strategy_hunyuan.json
export MODEL_BASE=data/hunyuan
rel_l1_thresh=0.15
CUDA_VISIBLE_DEVICES=0,1,2,3 torchrun --nnodes=1 --nproc_per_node=$num_gpus --master_port 29603 \
fastvideo/sample/sample_t2v_hunyuan_STA.py \
--height 768 \
--width 1280 \
--num_frames 117 \
--num_inference_steps 50 \
--guidance_scale 1 \
--embedded_cfg_scale 6 \
--flow_shift 7 \
--flow-reverse \
--prompt ./assets/prompt.txt \
--seed 12345 \
--output_path outputs_video/hunyuan_STA/ \
--model_path $MODEL_BASE \
--mask_strategy_file_path $mask_strategy_file_path \
--rel_l1_thresh $rel_l1_thresh \
--dit-weight ${MODEL_BASE}/hunyuan-video-t2v-720p/transformers/mp_rank_00_model_states.pt \
--vae-sp \
--enable_teacache
# Inference with STA only
num_gpus=1
mask_strategy_file_path=assets/mask_strategy_hunyuan.json
export MODEL_BASE=data/hunyuan
CUDA_VISIBLE_DEVICES=1 torchrun --nnodes=1 --nproc_per_node=$num_gpus --master_port 29603 \
fastvideo/sample/sample_t2v_hunyuan_STA.py \
--height 768 \
--width 1280 \
--num_frames 117 \
--num_inference_steps 50 \
--guidance_scale 1 \
--embedded_cfg_scale 6 \
--flow_shift 7 \
--flow-reverse \
--prompt ./assets/prompt.txt \
--seed 12345 \
--output_path outputs_video/hunyuan_STA/ \
--model_path $MODEL_BASE \
--mask_strategy_file_path $mask_strategy_file_path \
--dit-weight ${MODEL_BASE}/hunyuan-video-t2v-720p/transformers/mp_rank_00_model_states.pt \
--vae-sp \
--enable_torch_compile
scripts/inference/v0/inference_hunyuan_hf.sh
-15
View File
@@ -1,15 +0,0 @@
#!/bin/bash
num_gpus=4
torchrun --nnodes=1 --nproc_per_node=$num_gpus --master_port 29503 \
fastvideo/sample/sample_t2v_hunyuan_hf.py \
--model_path ~/data/hunyuan_diffusers/ \
--prompt_path "assets/prompt.txt" \
--num_frames 125 \
--height 720 \
--width 1280 \
--num_inference_steps 50 \
--output_path outputs_video/hunyuan_hf/ \
--seed 1024 \
scripts/inference/v0/inference_hunyuan_hf_quantization.sh
-19
View File
@@ -1,19 +0,0 @@
#!/bin/bash
num_gpus=1
export MODEL_BASE="data/FastHunyuan-diffusers"
torchrun --nnodes=1 --nproc_per_node=$num_gpus --master_port 12345 \
fastvideo/sample/sample_t2v_hunyuan_hf.py \
--height 720 \
--width 1280 \
--num_frames 45 \
--num_inference_steps 6 \
--guidance_scale 1 \
--embedded_cfg_scale 6 \
--flow_shift 17 \
--prompt ./assets/prompt.txt \
--seed 1024 \
--output_path outputs_video/hunyuan_quant/nf4/ \
--model_path $MODEL_BASE \
--quantization "nf4" \
--cpu_offload
scripts/inference/v0/inference_mochi_hf.sh
-18
View File
@@ -1,18 +0,0 @@
#!/bin/bash
num_gpus=4
torchrun --nnodes=1 --nproc_per_node=$num_gpus --master_port 29503 \
fastvideo/sample/sample_t2v_mochi.py \
--model_path data/FastMochi-diffusers \
--prompt_path "assets/prompt.txt" \
--num_frames 163 \
--height 480 \
--width 848 \
--num_inference_steps 8 \
--guidance_scale 1.5 \
--output_path outputs_video/mochi_hf/ \
--seed 1024 \
--scheduler_type "pcm_linear_quadratic" \
--linear_threshold 0.1 \
--linear_range 0.75
scripts/inference/v0/inference_stepvideo.sh
-19
View File
@@ -1,19 +0,0 @@
#!/bin/bash
# You better have two terminal, one for the remote server, and one for DiT
CUDA_VISIBLE_DEVICES=7 python fastvideo/sample/call_remote_server_stepvideo.py --model_dir data/stepvideo-t2v/ &
parallel=4
url='127.0.0.1'
model_dir=data/stepvideo-t2v
torchrun --nproc_per_node $parallel fastvideo/sample/sample_t2v_stepvideo.py \
--model_dir $model_dir \
--vae_url $url \
--caption_url $url \
--prompt assets/prompt.txt \
--infer_steps 50 \
--width 768 \
--height 768 \
--num_frames 204 \
--cfg_scale 9.0 \
--save_path outputs/ \
--time_shift 13.0
scripts/inference/v0/inference_stepvideo_STA.sh
-25
View File
@@ -1,25 +0,0 @@
#!/bin/bash
# Inference with STA + Teacache
# You better have two terminal, one for the remote server, and one for DiT
# remove enable_teacache to disable Teacache
CUDA_VISIBLE_DEVICES=7 python fastvideo/sample/call_remote_server_stepvideo.py --model_dir data/stepvideo-t2v/ &
parallel=4
url='127.0.0.1'
model_dir=data/stepvideo-t2v
mask_strategy_file_path=assets/mask_strategy_stepvideo.json
rel_l1_thresh=0.23
torchrun --nproc_per_node $parallel fastvideo/sample/sample_t2v_stepvideo_STA.py \
--model_dir $model_dir \
--vae_url $url \
--caption_url $url \
--prompt "A rocket blasting off from the launch pad, accelerating rapidly into the sky." \
--infer_steps 50 \
--width 768 \
--height 768 \
--num_frames 204 \
--cfg_scale 9.0 \
--save_path outputs/ \
--time_shift 13.0 \
--rel_l1_thresh $rel_l1_thresh \
--enable_teacache \
--mask_strategy_file_path $mask_strategy_file_path