Update entrypoint

gabrielmbmb · Sep 11, 2024 · 0c7af7a · 0c7af7a
1 parent 5c41161
commit 0c7af7a
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/candle-holder-serve/entrypoint.sh b/candle-holder-serve/entrypoint.sh
@@ -23,6 +23,10 @@ if [ ! -z "$CANDLE_HOLDER_DTYPE" ]; then
   ARGS+=("--dtype" "$CANDLE_HOLDER_DTYPE")
 fi
 
+if [ ! -z "$CANDLE_HOLDER_NUM_WORKERS" ]; then
+  ARGS+=("--num-workers" "$CANDLE_HOLDER_NUM_WORKERS")
+fi
+
 CANDLE_HOLDER_HOST=${CANDLE_HOLDER_HOST:-0.0.0.0:8080}
 ARGS+=("--host" "$CANDLE_HOLDER_HOST")