doanh25032004's picture
Backup source tree of video_gen_physics (2026-07-31T14:21:08Z)
ec0a9aa verified
Raw
History Blame Contribute Delete
947 Bytes
#!/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}" \
"$@"