Update docker-compose.yml

Author: devnen

docker-compose.yml

@@ -8,18 +8,40 @@ services:
     ports:
       - "${PORT:-8003}:${PORT:-8003}"
     volumes:
+      # Mount local directories into the container for persistent data
       - ./model_cache:/app/model_cache
       - ./reference_audio:/app/reference_audio
       - ./outputs:/app/outputs
       - ./voices:/app/voices
-      - ./config.yaml:/app/config.yaml
-    deploy:
-      resources:
-        reservations:
-          devices:
-            - driver: nvidia
-              count: 1
-              capabilities: [gpu]
+      # DO NOT mount config.yaml - let the app create it inside
+
+    # --- GPU Access ---
+    # Modern method (Recommended for newer Docker/NVIDIA setups)
+    devices:
+      - nvidia.com/gpu=all
+    device_cgroup_rules:
+      - "c 195:* rmw" # Needed for some NVIDIA container toolkit versions
+      - "c 236:* rmw" # Needed for some NVIDIA container toolkit versions
+
+    # Legacy method (Alternative for older Docker/NVIDIA setups)
+    # If the 'devices' block above doesn't work, comment it out and uncomment
+    # the 'deploy' block below. Do not use both simultaneously.
+    # deploy:
+    #   resources:
+    #     reservations:
+    #       devices:
+    #         - driver: nvidia
+    #           count: 1 # Or specify specific GPUs e.g., "device=0,1"
+    #           capabilities: [gpu]
+    # --- End GPU Access ---
+
     restart: unless-stopped
     env_file:
-      - .env
+      # Load environment variables from .env file for initial config seeding
+      - .env
+    environment:
+      # Enable faster Hugging Face downloads inside the container
+      - HF_HUB_ENABLE_HF_TRANSFER=1
+      # Pass GPU capabilities (may be needed for legacy method if uncommented)
+      - NVIDIA_VISIBLE_DEVICES=all
+      - NVIDIA_DRIVER_CAPABILITIES=compute,utility