a.pourmand commited on
Commit
39e129a
1 Parent(s): 6a51822
Files changed (1) hide show
  1. docker-compose.yaml +30 -0
docker-compose.yaml ADDED
@@ -0,0 +1,30 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ version: "3.7"
2
+
3
+ services:
4
+ seamless_diarization_VAD_service:
5
+ image: seamless_diarization_VAD:v0.1
6
+ build:
7
+ context: .
8
+ container_name: seamless_diarization_VAD
9
+ user: root
10
+ environment:
11
+ - GIT_PYTHON_REFRESH=quiet
12
+ - HUGGINGFACE_HUB_CACHE=/home/user/temp/.cache/HUGGINGFACE_HUB_CACHE
13
+ - TRANSFORMERS_CACHE=/home/user/temp/.cache/TRANSFORMERS_CACHE
14
+ - HF_HOME=/home/user/temp/.cache/HF_HOME
15
+ - TORCH_HOME=/home/user/temp/torch
16
+ #- TRANSFORMERS_OFFLINE=1
17
+ tty: true
18
+ stdin_open: true
19
+ volumes:
20
+ - ../torch/:/home/user/temp/torch
21
+ - ./cache/:/home/user/.cache/
22
+ ports:
23
+ - 8005:9000
24
+ deploy:
25
+ resources:
26
+ reservations:
27
+ devices:
28
+ - driver: nvidia
29
+ device_ids: ["4"]
30
+ capabilities: [gpu]