Spaces:
Running
on
Zero
Running
on
Zero
File size: 889 Bytes
96257b2 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 |
#!/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
|