#!/usr/bin/env bash
set -euo pipefail
mkdir -p /workspace/agnes/code
python -m pip install --break-system-packages --no-cache-dir uv
uv venv --system-site-packages /workspace/agnes/venv
export PATH="/workspace/agnes/venv/bin:$PATH"
uv pip install --python /workspace/agnes/venv/bin/python \
  'torch==2.13.0' 'transformers==5.14.0' 'accelerate==1.12.0' \
  'huggingface_hub==1.18.0' 'pillow>=12' 'sentencepiece>=0.2' 'einops' \
  'nvidia-modelopt @ https://github.com/NVIDIA/Model-Optimizer/archive/5cae3940402f1ced98069a666b0bec72ec8b33b5.tar.gz'
python - <<'PY'
import os
from huggingface_hub import hf_hub_download
hf_hub_download('ProCreations/Agnes-3.0-Flash-NVFP4','reproducibility/build.py',
    revision=os.environ['AGNES_CODE_REVISION'],local_dir='/workspace/agnes/code')
PY
python -u /workspace/agnes/code/reproducibility/build.py