Skip to content

Commit 6cadb51

Browse files
Update config.json
1 parent 7a0bbbf commit 6cadb51

File tree

1 file changed

+12
-9
lines changed

1 file changed

+12
-9
lines changed

config.json

Lines changed: 12 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -5,8 +5,9 @@
55
"modelBasePath": "src/asr/whisper_en/local/.",
66
"apiBasePath": "asr/whisper_en/local/",
77
"containerPort": 8000,
8-
"environment": {},
9-
"nginx": [],
8+
"environment": {
9+
"NVIDIA_VISIBLE_DEVICES": "${NVIDIA_VISIBLE_DEVICES}"
10+
},
1011
"nginx": ["client_max_body_size 100M;", "proxy_read_timeout 600;", "proxy_connect_timeout 600;", "proxy_send_timeout 600;"],
1112
"constraints": ["node.labels.node_vm_type==gpu"],
1213
"build": false
@@ -16,8 +17,9 @@
1617
"modelBasePath": "src/asr/whisper_lang_rec/local/.",
1718
"apiBasePath": "asr/whisper_lang_rec/local/",
1819
"containerPort": 8000,
19-
"environment": {},
20-
"nginx": [],
20+
"environment": {
21+
"NVIDIA_VISIBLE_DEVICES": "${NVIDIA_VISIBLE_DEVICES}"
22+
},
2123
"nginx": ["client_max_body_size 100M;", "proxy_read_timeout 600;", "proxy_connect_timeout 600;", "proxy_send_timeout 600;"],
2224
"constraints": ["node.labels.node_vm_type==gpu"],
2325
"build": false
@@ -27,8 +29,11 @@
2729
"modelBasePath": "src/ner/agri_ner_akai/local/.",
2830
"apiBasePath": "ner/agri_ner_akai/local/",
2931
"containerPort": 8000,
30-
"environment": {},
32+
"environment": {
33+
"NVIDIA_VISIBLE_DEVICES": "${NVIDIA_VISIBLE_DEVICES}"
34+
},
3135
"nginx": [],
36+
"constraints": ["node.labels.node_vm_type==gpu"],
3237
"build": false
3338
},
3439
{
@@ -57,7 +62,7 @@
5762
"environment": {},
5863
"nginx": [],
5964
"constraints": ["node.role==worker"],
60-
"build": false
65+
"build": true
6166
},
6267
{
6368
"serviceName": "text_translation_azure_dict",
@@ -76,11 +81,9 @@
7681
"apiBasePath": "data_generation/dictionary_aug/remote/",
7782
"containerPort": 8000,
7883
"environment": {
79-
"OPENAI_API_KEY": "${OPENAI_API_KEY}",
80-
"NVIDIA_VISIBLE_DEVICES": "${NVIDIA_VISIBLE_DEVICES}"
84+
"OPENAI_API_KEY": "${OPENAI_API_KEY}"
8185
},
8286
"nginx": [],
83-
"constraints": ["node.labels.node_vm_type==gpu"],
8487
"build": false
8588
},
8689
{

0 commit comments

Comments
 (0)