|
5 | 5 | "modelBasePath": "src/asr/whisper_en/local/.", |
6 | 6 | "apiBasePath": "asr/whisper_en/local/", |
7 | 7 | "containerPort": 8000, |
8 | | - "environment": {}, |
9 | | - "nginx": [], |
| 8 | + "environment": { |
| 9 | + "NVIDIA_VISIBLE_DEVICES": "${NVIDIA_VISIBLE_DEVICES}" |
| 10 | + }, |
10 | 11 | "nginx": ["client_max_body_size 100M;", "proxy_read_timeout 600;", "proxy_connect_timeout 600;", "proxy_send_timeout 600;"], |
11 | 12 | "constraints": ["node.labels.node_vm_type==gpu"], |
12 | 13 | "build": false |
|
16 | 17 | "modelBasePath": "src/asr/whisper_lang_rec/local/.", |
17 | 18 | "apiBasePath": "asr/whisper_lang_rec/local/", |
18 | 19 | "containerPort": 8000, |
19 | | - "environment": {}, |
20 | | - "nginx": [], |
| 20 | + "environment": { |
| 21 | + "NVIDIA_VISIBLE_DEVICES": "${NVIDIA_VISIBLE_DEVICES}" |
| 22 | + }, |
21 | 23 | "nginx": ["client_max_body_size 100M;", "proxy_read_timeout 600;", "proxy_connect_timeout 600;", "proxy_send_timeout 600;"], |
22 | 24 | "constraints": ["node.labels.node_vm_type==gpu"], |
23 | 25 | "build": false |
|
27 | 29 | "modelBasePath": "src/ner/agri_ner_akai/local/.", |
28 | 30 | "apiBasePath": "ner/agri_ner_akai/local/", |
29 | 31 | "containerPort": 8000, |
30 | | - "environment": {}, |
| 32 | + "environment": { |
| 33 | + "NVIDIA_VISIBLE_DEVICES": "${NVIDIA_VISIBLE_DEVICES}" |
| 34 | + }, |
31 | 35 | "nginx": [], |
| 36 | + "constraints": ["node.labels.node_vm_type==gpu"], |
32 | 37 | "build": false |
33 | 38 | }, |
34 | 39 | { |
|
57 | 62 | "environment": {}, |
58 | 63 | "nginx": [], |
59 | 64 | "constraints": ["node.role==worker"], |
60 | | - "build": false |
| 65 | + "build": true |
61 | 66 | }, |
62 | 67 | { |
63 | 68 | "serviceName": "text_translation_azure_dict", |
|
76 | 81 | "apiBasePath": "data_generation/dictionary_aug/remote/", |
77 | 82 | "containerPort": 8000, |
78 | 83 | "environment": { |
79 | | - "OPENAI_API_KEY": "${OPENAI_API_KEY}", |
80 | | - "NVIDIA_VISIBLE_DEVICES": "${NVIDIA_VISIBLE_DEVICES}" |
| 84 | + "OPENAI_API_KEY": "${OPENAI_API_KEY}" |
81 | 85 | }, |
82 | 86 | "nginx": [], |
83 | | - "constraints": ["node.labels.node_vm_type==gpu"], |
84 | 87 | "build": false |
85 | 88 | }, |
86 | 89 | { |
|
0 commit comments