|
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 |
|
|
|
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] |
|
|