J Z commited on
Commit
91e04d7
·
verified ·
1 Parent(s): b444f1c

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +6 -0
Dockerfile CHANGED
@@ -44,6 +44,12 @@ RUN pip install --no-cache-dir \
44
  # Create dirs
45
  RUN mkdir -p /tmp/videos /tmp/tts_output /tmp/lipsync_output
46
 
 
 
 
 
 
 
47
  # ---- APP CODE (only this rebuilds on code changes) ----
48
  COPY --chown=user ./app.py ./llm.py ./tts.py ./float_lipsync.py /app/
49
 
 
44
  # Create dirs
45
  RUN mkdir -p /tmp/videos /tmp/tts_output /tmp/lipsync_output
46
 
47
+ # Pre-download Orpheus TTS model + SNAC decoder (cached in Docker layer)
48
+ RUN python -c "from huggingface_hub import snapshot_download; \
49
+ snapshot_download('canopylabs/orpheus-3b-0.1-pretrained', ignore_patterns=['*.md','*.txt']); \
50
+ snapshot_download('hubertsiuzdak/snac_24khz')" \
51
+ || echo 'WARNING: Model pre-download failed, will retry at runtime'
52
+
53
  # ---- APP CODE (only this rebuilds on code changes) ----
54
  COPY --chown=user ./app.py ./llm.py ./tts.py ./float_lipsync.py /app/
55