Nekochu commited on
Commit
c37e80e
·
1 Parent(s): f59d542

pre-download training checkpoints at build time

Browse files
Files changed (1) hide show
  1. 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