change

docker-compose.yaml

@@ -0,0 +1,30 @@
+version: "3.7"
+
+services:
+  seamless_diarization_VAD_service:
+    image: seamless_diarization_VAD:v0.1
+    build:
+      context: .
+    container_name: seamless_diarization_VAD
+    user: root
+    environment:
+      - GIT_PYTHON_REFRESH=quiet
+      - HUGGINGFACE_HUB_CACHE=/home/user/temp/.cache/HUGGINGFACE_HUB_CACHE
+      - TRANSFORMERS_CACHE=/home/user/temp/.cache/TRANSFORMERS_CACHE
+      - HF_HOME=/home/user/temp/.cache/HF_HOME
+      - TORCH_HOME=/home/user/temp/torch
+      #- TRANSFORMERS_OFFLINE=1
+    tty: true
+    stdin_open: true
+    volumes:
+      - ../torch/:/home/user/temp/torch
+      - ./cache/:/home/user/.cache/
+    ports:
+      - 8005:9000
+    deploy:
+      resources:
+        reservations:
+          devices:
+            - driver: nvidia
+              device_ids: ["4"]
+              capabilities: [gpu]