| @@ | |
| FROM pytorch/pytorch:2.6.0-cuda12.6-cudnn9-devel | |
| #––––––––––––––––––––––––––––––––––––––––––––––––––––––––––––––––– | |
| # 1. Hugging Face cache lives in /data (.hf Space volume) * | |
| #––––––––––––––––––––––––––––––––––––––––––––––––––––––––––––––––– | |
| ENV HF_HOME=/data/.huggingface | |
| @@ | |
| WORKDIR /workspace/spatial-dia | |
| ENV PYTHONUNBUFFERED=1 | |
| CMD ["/bin/bash"] | |
| #––––––––––––––––––––––––––––––––––––––––––––––––––––––––––––––––– | |
| # 2. Boot script: pre-fetch weights once, then launch Gradio | |
| #––––––––––––––––––––––––––––––––––––––––––––––––––––––––––––––––– | |
| COPY entrypoint.sh /entrypoint.sh | |
| RUN chmod +x /entrypoint.sh | |
| CMD ["/entrypoint.sh"] | |