File size: 947 Bytes
ec0a9aa | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 | #!/usr/bin/env bash
set -euo pipefail
REPO_ROOT="$(cd "$(dirname "${BASH_SOURCE[0]}")/../.." && pwd)"
cd "${REPO_ROOT}"
if [[ -f "${REPO_ROOT}/models/DreamDojo/.venv/bin/activate" ]]; then
source "${REPO_ROOT}/models/DreamDojo/.venv/bin/activate"
fi
PROCESSED_ROOT="${PROCESSED_ROOT:-datasets/processed/ctrl_world_robo_multiview}"
SVD_MODEL_PATH="${SVD_MODEL_PATH:-checkpoints/stabilityai/stable-video-diffusion-img2vid}"
CLIP_MODEL_PATH="${CLIP_MODEL_PATH:-checkpoints/openai/clip-vit-base-patch32}"
OUTPUT_DIR="${OUTPUT_DIR:-${PROCESSED_ROOT}/checkpoints}"
CKPT="${CKPT:-${OUTPUT_DIR}/latest.pt}"
NUM_SAMPLES="${NUM_SAMPLES:-8}"
python -m finetuning.eval_replay \
--processed-root "${PROCESSED_ROOT}" \
--manifest-path "${PROCESSED_ROOT}/manifest.json" \
--svd-model-path "${SVD_MODEL_PATH}" \
--clip-model-path "${CLIP_MODEL_PATH}" \
--output-dir "${OUTPUT_DIR}" \
--ckpt "${CKPT}" \
--num-samples "${NUM_SAMPLES}" \
"$@"
|