Zhang Peiyuan
d417e4c7c4
typo add upload command add add ema_transform add ema transformer remove harcode ok ema remove hardcode update env update script; no sp revert to sp=4, sp bs=2, full shard add aws efo env distributed validation readme distributed validation add gupload linear range; fix uncond; multi guidance validation add script
13 lines
324 B
Bash
13 lines
324 B
Bash
num_gpus=1
|
|
|
|
torchrun --nproc_per_node=$num_gpus fastvideo/sample/generate_synthetic.py \
|
|
--model_path data/mochi \
|
|
--num_frames 1 \
|
|
--height 480 \
|
|
--width 848 \
|
|
--num_inference_steps 4 \
|
|
--guidance_scale 4.5 \
|
|
--prompt_path "data/prompt.txt" \
|
|
--dataset_output_dir data/synthetic_debug2
|
|
|
|
|