| set -euo pipefail | |
| ROOT="$(cd "$(dirname "${BASH_SOURCE[0]}")/.." && pwd)" | |
| : "${INPUT_DIR:?Set INPUT_DIR to a directory of input videos}" | |
| : "${MASK_DIR:?Set MASK_DIR to a directory of same-named mask videos}" | |
| OUTPUT_DIR="${OUTPUT_DIR:-$ROOT/outputs/dmd_2step}" | |
| GPU="${GPU:-0}" | |
| PYTHON_BIN="${PYTHON_BIN:-python}" | |
| cd "$ROOT/src" | |
| CUDA_VISIBLE_DEVICES="$GPU" "$PYTHON_BIN" -u infer.py \ | |
| --model_name "$ROOT/checkpoints/dmd_2step" \ | |
| --common_model_name "$ROOT/checkpoints/common" \ | |
| --config_path "$ROOT/configs/wan2.1/wan_civitai.yaml" \ | |
| --lightvae_path "$ROOT/checkpoints/common/lightvae.pth" \ | |
| --skip_lora --dmd_steps 2 --guidance_scale 1.0 \ | |
| --input_dir "$INPUT_DIR" --input_mask_dir "$MASK_DIR" --save_dir "$OUTPUT_DIR" | |