#!/usr/bin/env bash
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"