diff --git a/containers/chatterbox/chatterbox.Containerfile b/containers/chatterbox/chatterbox.Containerfile index 9bb88b3..584f8ae 100644 --- a/containers/chatterbox/chatterbox.Containerfile +++ b/containers/chatterbox/chatterbox.Containerfile @@ -26,8 +26,12 @@ RUN if [ "$DEVICE" = "cpu" ]; then \ pip install --no-cache-dir torch torchaudio --index-url https://download.pytorch.org/whl/cpu; \ fi -# Install chatterbox-tts and a minimal API server -RUN pip install --no-cache-dir chatterbox-tts fastapi uvicorn python-multipart soundfile +RUN pip install --no-cache-dir --no-deps chatterbox-tts \ + && pip install --no-cache-dir \ + numpy librosa==0.11.0 s3tokenizer transformers==5.2.0 diffusers==0.29.0 \ + resemble-perth conformer==0.3.2 safetensors==0.5.3 spacy-pkuseg \ + pykakasi==2.3.0 gradio==6.8.0 pyloudnorm omegaconf \ + fastapi uvicorn python-multipart soundfile # Models download at runtime here — mount as a volume to persist them RUN mkdir -p /app/models