PusaV1 / demos /cli_test_ti2v_release.sh
rahul7star's picture
Migrated from GitHub
96257b2 verified
#!/bin/bash
# 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