19 lines
600 B
Bash
19 lines
600 B
Bash
#!/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 |