| 1 | services: |
| 2 | tts: |
| 3 | image: soar97/triton-f5-tts:24.12 |
| 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 "pip install vocos && rm -rf F5-TTS && git clone https://github.com/SWivid/F5-TTS.git && cd F5-TTS/src/f5_tts/runtime/triton_trtllm/ && bash run.sh 0 4 $MODEL" |
| 21 |