45 lines
2.0 KiB
Bash
45 lines
2.0 KiB
Bash
#!/usr/bin/env bash
|
|
# Stop at the cheapest failing boundary. Run inside a prepared GPU environment.
|
|
set -euo pipefail
|
|
cd "$(dirname "${BASH_SOURCE[0]}")/.."
|
|
|
|
component=${1:-all}
|
|
level=${2:-golden}
|
|
case "$component" in all|vae|dense|causal) ;; *) echo 'component: all | vae | dense | causal' >&2; exit 2;; esac
|
|
case "$level" in golden|parity|default) ;; *) echo 'level: golden | parity | default' >&2; exit 2;; esac
|
|
|
|
pytest fastvideo/tests/api/test_wan_definitions.py \
|
|
fastvideo/tests/loader/test_wan_family_imports.py \
|
|
fastvideo/tests/stages/test_cfg_gating.py fastvideo/tests/stages/test_wan_denoising.py \
|
|
fastvideo/tests/stages/test_wan_dmd_denoising.py fastvideo/tests/stages/test_wan_causal_denoising.py \
|
|
fastvideo/tests/stages/test_wan_pipeline_wiring.py -xq
|
|
|
|
if [[ $component == all || $component == vae ]]; then
|
|
pytest fastvideo/tests/golden_gate/test_wan_vae.py -xq
|
|
fi
|
|
if [[ $component == all || $component == dense ]]; then
|
|
pytest fastvideo/tests/golden_gate/test_wan_t2v.py -xq
|
|
pytest fastvideo/tests/golden_gate/test_wan_denoising.py -xq
|
|
fi
|
|
if [[ $component == all || $component == causal ]]; then
|
|
pytest fastvideo/tests/golden_gate/test_wan_causal.py -xq
|
|
fi
|
|
[[ $level == golden ]] && exit 0
|
|
|
|
if [[ $component == all || $component == vae ]]; then
|
|
pytest fastvideo/tests/vaes/test_wan_vae.py -xq
|
|
fi
|
|
if [[ $component == all || $component == dense ]]; then
|
|
pytest fastvideo/tests/transformers/test_wanvideo.py -xq
|
|
fi
|
|
[[ $level == parity ]] && exit 0
|
|
|
|
# Preserve the selected model, prompt, backend, device and reference tier. The
|
|
# caller supplies matching SSIM environment settings; this command does not
|
|
# rewrite references or silently downgrade full-quality requests.
|
|
case "$component" in
|
|
vae|all) pytest fastvideo/tests/ssim/test_wan_t2v_similarity.py fastvideo/tests/ssim/test_wan_i2v_similarity.py fastvideo/tests/ssim/test_causal_similarity.py -xq;;
|
|
dense) pytest fastvideo/tests/ssim/test_wan_t2v_similarity.py fastvideo/tests/ssim/test_wan_i2v_similarity.py -xq;;
|
|
causal) pytest fastvideo/tests/ssim/test_causal_similarity.py -xq;;
|
|
esac
|