Update Docker Compose for TTS service configuration

Author: webadmin2021

src/f5_tts/runtime/triton_trtllm/docker-compose.yml

@@ -1,7 +1,7 @@
 services:
   tts:
     image: soar97/triton-f5-tts:24.12
-    shm_size: '1gb'
+    shm_size: '2gb'
     ports:
       - "8000:8000"
       - "8001:8001"
@@ -14,7 +14,7 @@ services:
         reservations:
           devices:
             - driver: nvidia
-              device_ids: ['0']
+              device_ids: ['0', '1']
               capabilities: [gpu]
     command: >
-      /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"
+      /bin/bash -c "pip install vocos rjieba && rm -rf F5-TTS && git clone --branch multi-gpu https://github.com/webadmin2021/F5-TTS.git F5-TTS && cd F5-TTS/src/f5_tts/runtime/triton_trtllm/ && bash run.sh 0 4 $MODEL"