#!/usr/bin/env bash
# Runs on the pod: Python deps + model downloads into /workspace (idempotent).
set -euo pipefail
cd /workspace
export HF_HUB_ENABLE_HF_TRANSFER=0 HF_XET_HIGH_PERFORMANCE=1 PIP_BREAK_SYSTEM_PACKAGES=1
pip install -q --root-user-action=ignore "diffusers==0.38.0" "transformers==5.17.0" "accelerate>=1.10" \
    "safetensors>=0.6" "huggingface_hub[hf_xet]>=1.0" pillow flash-linear-attention 2>&1 | tail -3
python - <<'EOF'
import torch, diffusers, transformers
print('torch', torch.__version__, 'cuda', torch.version.cuda, 'diffusers', diffusers.__version__,
      'transformers', transformers.__version__, [torch.cuda.get_device_name(i) for i in range(torch.cuda.device_count())])
EOF
mkdir -p models
hf download OpenMOSS-Team/MOSS-Image-SFT --local-dir models/MOSS-Image-SFT --exclude "assets/*" > /dev/null
hf download lodestones/MOSS-Image-SFT-8-steps-dmd lora.safetensors time_embedding.safetensors dataspace.safetensors README.md \
    --local-dir models/MOSS-Image-SFT-8-steps-dmd > /dev/null
du -sh models/*
md5sum models/MOSS-Image-SFT/pipeline_moss_image.py