Spaces:
Running
Running
pre-download training checkpoints at build time
Browse files- Dockerfile +5 -0
Dockerfile
CHANGED
|
@@ -78,6 +78,11 @@ RUN pip3 install --no-cache-dir --extra-index-url https://download.pytorch.org/w
|
|
| 78 |
# Clone ACE-Step repo for training module
|
| 79 |
RUN git clone --depth 1 https://github.com/ace-step/ACE-Step-1.5 /app/ace-step-source
|
| 80 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 81 |
# Copy application files
|
| 82 |
COPY app.py /app/app.py
|
| 83 |
COPY start.sh /app/start.sh
|
|
|
|
| 78 |
# Clone ACE-Step repo for training module
|
| 79 |
RUN git clone --depth 1 https://github.com/ace-step/ACE-Step-1.5 /app/ace-step-source
|
| 80 |
|
| 81 |
+
# Pre-download training checkpoints (avoids 10GB runtime download)
|
| 82 |
+
RUN python3 -c "from huggingface_hub import snapshot_download; \
|
| 83 |
+
snapshot_download('ACE-Step/Ace-Step1.5', local_dir='/app/checkpoints', \
|
| 84 |
+
ignore_patterns=['*.md', '*.txt', '.gitattributes'])"
|
| 85 |
+
|
| 86 |
# Copy application files
|
| 87 |
COPY app.py /app/app.py
|
| 88 |
COPY start.sh /app/start.sh
|