#!/usr/bin/env bash
# Copy the dssd inference package into this Space (single source: BranchyLLM, branch dssd-v2/dssd-pub).
# Usage: scripts/vendor_dssd.sh [path/to/BranchyLLM checkout]
set -euo pipefail
SRC=${1:-../BranchyLLM-pub}
DST=$(cd "$(dirname "$0")/.." && pwd)/dssd
rm -rf "$DST"
mkdir -p "$DST/calibration"
for f in __init__ api decode fast heads hub kernels layered pld stops; do cp "$SRC/dssd/$f.py" "$DST/"; done
cp "$SRC/dssd/calibration/__init__.py" "$SRC/dssd/calibration/online.py" "$SRC/dssd/calibration/calib.py" "$DST/calibration/"
echo "Vendored from BranchyLLM $(git -C "$SRC" rev-parse --short HEAD) ($(git -C "$SRC" branch --show-current)); do not edit here." > "$DST/VENDORED.txt"
cat "$DST/VENDORED.txt"