Spaces:
Running
on
Zero
Running
on
Zero
# Model directory | |
MODEL_DIR="<path_to_Pusa-V0.5_dir>" | |
# Checkpoint paths | |
CHECKPOINTS=( | |
"<path_to_pusa_dit_checkpoint>" | |
) | |
# I2V - default conditioned on the first frame | |
image_dir="<path_to_input_image_dir>" | |
# prompt_dir=None | |
prompt="<your_prompt_here>" | |
# Strongly suggest to try different con_position here and also you can modify the level of noise add to the condition image, you may find some surprise | |
cond_position=0 | |
num_steps=30 | |
noise_multiplier=0.4 | |
# Generate with 1 GPU, generated video will be splitted into 4 parts for decoding | |
for CHECKPOINT in "${CHECKPOINTS[@]}"; do | |
CUDA_VISIBLE_DEVICES=6 python3 ./demos/cli_test_ti2v_release.py --model_dir "$MODEL_DIR" --dit_path "$CHECKPOINT" --image_dir "$image_dir" --prompt "$prompt" --prompt_dir "$prompt_dir" --cond_position "$cond_position" --num_steps "$num_steps" --noise_multiplier "$noise_multiplier" | |
done | |