Update config.json

AmakrushAI · Oct 7, 2023 · 6cadb51 · 6cadb51
1 parent 7a0bbbf
commit 6cadb51
Showing 1 changed file with 12 additions and 9 deletions.
diff --git a/config.json b/config.json
@@ -5,8 +5,9 @@
       "modelBasePath": "src/asr/whisper_en/local/.",
       "apiBasePath": "asr/whisper_en/local/",
       "containerPort": 8000,
-      "environment": {},
-      "nginx": [],
+       "environment": {
+        "NVIDIA_VISIBLE_DEVICES": "${NVIDIA_VISIBLE_DEVICES}"
+      },
       "nginx": ["client_max_body_size 100M;", "proxy_read_timeout 600;", "proxy_connect_timeout 600;", "proxy_send_timeout 600;"],
       "constraints": ["node.labels.node_vm_type==gpu"],
       "build": false
@@ -16,8 +17,9 @@
       "modelBasePath": "src/asr/whisper_lang_rec/local/.",
       "apiBasePath": "asr/whisper_lang_rec/local/",
       "containerPort": 8000,
-      "environment": {},
-      "nginx": [],
+      "environment": {
+        "NVIDIA_VISIBLE_DEVICES": "${NVIDIA_VISIBLE_DEVICES}"
+      },     
       "nginx": ["client_max_body_size 100M;", "proxy_read_timeout 600;", "proxy_connect_timeout 600;", "proxy_send_timeout 600;"],
       "constraints": ["node.labels.node_vm_type==gpu"],
       "build": false
@@ -27,8 +29,11 @@
       "modelBasePath": "src/ner/agri_ner_akai/local/.",
       "apiBasePath": "ner/agri_ner_akai/local/",
       "containerPort": 8000,
-      "environment": {},
+       "environment": {
+        "NVIDIA_VISIBLE_DEVICES": "${NVIDIA_VISIBLE_DEVICES}"
+      }, 
       "nginx": [],
+      "constraints": ["node.labels.node_vm_type==gpu"],
       "build": false
     },    
         {
@@ -57,7 +62,7 @@
       "environment": {},
       "nginx": [],
       "constraints": ["node.role==worker"],
-      "build": false
+      "build": true
     },
     {
       "serviceName": "text_translation_azure_dict",
@@ -76,11 +81,9 @@
       "apiBasePath": "data_generation/dictionary_aug/remote/",
       "containerPort": 8000,
       "environment": {
-        "OPENAI_API_KEY": "${OPENAI_API_KEY}",
-        "NVIDIA_VISIBLE_DEVICES": "${NVIDIA_VISIBLE_DEVICES}"
+        "OPENAI_API_KEY": "${OPENAI_API_KEY}"
       },
       "nginx": [],
-      "constraints": ["node.labels.node_vm_type==gpu"],
       "build": false
     },   
     {