File size: 822 Bytes
39e129a ce4d2b4 39e129a 53445d9 39e129a e684f26 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:7860
deploy:
resources:
reservations:
devices:
- driver: nvidia
device_ids: ["4"]
capabilities: [gpu]
|