#!/usr/bin/env bash
# Minimal Ctrl-World replay on bundled droid_subset.
# Use a GPU with enough free memory (~12GB+). Example: CUDA_VISIBLE_DEVICES=7
set -e
cd "$(dirname "$0")"
CKPT_ROOT="$(pwd)/checkpoints"
export CUDA_VISIBLE_DEVICES="${CUDA_VISIBLE_DEVICES:-0}"

python scripts/rollout_replay_traj.py \
  --dataset_root_path dataset_example \
  --dataset_meta_info_path dataset_meta_info \
  --dataset_names droid_subset \
  --data_stat_path dataset_meta_info/droid_subset/stat.json \
  --svd_model_path "$CKPT_ROOT/stable-video-diffusion-img2vid" \
  --clip_model_path "$CKPT_ROOT/clip-vit-base-patch32" \
  --ckpt_path "$CKPT_ROOT/ctrl-world/checkpoint-10000.pt"