Update start-workers.sh

ucsd-ets · Mar 5, 2024 · 7e6600a · 7e6600a
1 parent daa0746
commit 7e6600a
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/start-workers.sh b/start-workers.sh
@@ -7,13 +7,13 @@ if [ -x "$USERSCRIPT" ]; then
     exec "$USERSCRIPT" "$@"
 fi
 
-NUM_WORKERS=2
+NUM_WORKERS=${NUM_WORKERS:-2}
 
-WORKER_CPU_REQUEST=8
-WORKER_CPU_LIMIT=8
-WORKER_MEM_REQUEST=16384M
-WORKER_MEM_LIMIT=16384M
-WORKER_GPU_COUNT=0
+WORKER_CPU_REQUEST=${WORKER_CPU_REQUEST:-8}
+WORKER_CPU_LIMIT=${WORKER_CPU_LIMIT:-8}
+WORKER_MEM_REQUEST=${WORKER_MEM_REQUEST:-16384M}
+WORKER_MEM_LIMIT=${WORKER_MEM_LIMIT:-16384M}
+WORKER_GPU_COUNT=${WORKER_GPU_COUNT:-0}
 
 IMAGE=${JUPYTER_IMAGE_SPEC:-${DOCKER_IMAGE}}
 echo "STARTING WORKERS WITH WORKER_CPU_REQUEST=${WORKER_CPU_REQUEST}"