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:7860
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              device_ids: ["4"]
              capabilities: [gpu]