File size: 823 Bytes
39e129a
 
 
 
ce4d2b4
39e129a
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
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]