#!/usr/bin/env bash
set -euo pipefail

WEIGHT_DIR="${WEIGHT_DIR:-/path/to/weight/pretrained}"
mkdir -p "$WEIGHT_DIR"
curl -L --fail --retry 3 "https://huggingface.co/salv47/spherical-dyffusion/resolve/main/forecaster-sfno-best-inference_avg_crps.ckpt?download=true" -o "$WEIGHT_DIR/forecaster-sfno-best-inference_avg_crps.ckpt"
curl -L --fail --retry 3 "https://huggingface.co/salv47/spherical-dyffusion/resolve/main/interpolator-sfno-best-val_avg_crps.ckpt?download=true" -o "$WEIGHT_DIR/interpolator-sfno-best-val_avg_crps.ckpt"
curl -L --fail --retry 3 "https://huggingface.co/salv47/spherical-dyffusion/resolve/main/interpolator_sfno_paper_v0_hydra_config.yaml?download=true" -o "$WEIGHT_DIR/interpolator_sfno_paper_v0_hydra_config.yaml"