docker: bake soundfile, evaluate, tensorboard for notebook deps
The TTS notebooks (Sesame CSM, Orpheus) read audio via soundfile, the Whisper notebook computes WER via evaluate, and TrainingArguments defaults report_to to tensorboard. These are declared by notebook pip cells that the in-image notebook runner neutralises (deps are meant to be prebaked), so without them those notebooks die on import. All are pure-Python or self-contained wheels and never name torch, so the cu128 pin set is undisturbed.
This commit is contained in:
parent
2cd58d5f38
commit
5bb47cf3cb
1 changed files with 7 additions and 1 deletions
|
|
@ -272,9 +272,15 @@ RUN set -eux \
|
|||
# matplotlib rides along for the notebook crowd: plotting is table stakes in
|
||||
# a Jupyter image, and several model repos' trust_remote_code modeling files
|
||||
# (e.g. DeepSeek-OCR) import it unconditionally.
|
||||
# soundfile (TTS notebooks read/write audio; bundles libsndfile in its wheel),
|
||||
# evaluate (Whisper notebook's WER metric), and tensorboard (default
|
||||
# TrainingArguments report_to backend) are declared by notebook install cells
|
||||
# that the in-image runner neutralises, so bake them here. All pure-Python or
|
||||
# self-contained wheels; none names torch, so the cu128 pin is undisturbed.
|
||||
RUN ${VENV}/bin/uv pip install \
|
||||
--python ${VENV}/bin/python \
|
||||
jupyterlab notebook ipywidgets matplotlib
|
||||
jupyterlab notebook ipywidgets matplotlib \
|
||||
soundfile evaluate tensorboard
|
||||
|
||||
# Audio decode out of the box: the TTS/STT notebooks feed datasets' Audio
|
||||
# features, which decode through torchcodec. Three traps, all defended:
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue