docker-compose.yml 558 B

1234567891011121314151617181920
  1. services:
  2. tts:
  3. image: soar97/triton-spark-tts:25.02
  4. shm_size: '1gb'
  5. ports:
  6. - "8000:8000"
  7. - "8001:8001"
  8. - "8002:8002"
  9. environment:
  10. - PYTHONIOENCODING=utf-8
  11. - MODEL_ID=${MODEL_ID}
  12. deploy:
  13. resources:
  14. reservations:
  15. devices:
  16. - driver: nvidia
  17. device_ids: ['0']
  18. capabilities: [gpu]
  19. command: >
  20. /bin/bash -c "rm -rf Spark-TTS && git clone https://github.com/SparkAudio/Spark-TTS.git && cd Spark-TTS/runtime/triton_trtllm && bash run.sh 0 3"