Add configurable WORKERS env var, default to 1 for GPU models

This commit is contained in:
russell@unturf.com 2026-01-26 13:16:44 -05:00
parent f299b43f1a
commit 0159f1f216
3 changed files with 14 additions and 1 deletions

View file

@ -2,6 +2,9 @@
[ -f speech.env ] && . speech.env
# Default to 1 worker for GPU models (Qwen3-TTS)
WORKERS=${WORKERS:-1}
echo "First startup may download ~3GB of Qwen3-TTS model. Please wait."
# Pre-download Qwen3-TTS model (default engine)
@ -11,4 +14,4 @@ python -c "from qwen_tts import Qwen3TTSModel; Qwen3TTSModel.from_pretrained('Qw
# bash download_voices_tts-1.sh
# bash download_voices_tts-1-hd.sh $PRELOAD_MODEL
python speech.py ${PRELOAD_MODEL:+--preload $PRELOAD_MODEL} $EXTRA_ARGS $@
python speech.py --workers $WORKERS ${PRELOAD_MODEL:+--preload $PRELOAD_MODEL} $EXTRA_ARGS $@