PusaV1 / demos /cli_test_multi_frames_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>"
)
num_steps=30
# Define multiple conditioning inputs
MULTI_COND='<your_multi_cond_here>' # Format: {position: [image_path, prompt_path, noise_multiplier], ...}
# e.g., Start&End Frames Generation: MULTI_COND='{
# "0": ["./demos/example3.jpg", 0.3],
# "20": ["./demos/example4.jpg", 0.7]
# }'
prompt="<your_prompt_here>"
# Loop through each checkpoint and run the command with multiple conditioning
for CHECKPOINT in "${CHECKPOINTS[@]}"; do
CUDA_VISIBLE_DEVICES=0 python3 ./demos/cli_test_multi_frames_release.py \
--model_dir "$MODEL_DIR" \
--prompt "$prompt" \
--dit_path "$CHECKPOINT" \
--multi_cond "$MULTI_COND" \
--num_steps "$num_steps" \
done