0xarchit commited on
Commit
a6b5235
·
1 Parent(s): b3c9994

Add REASONING env toggle for llama-server reasoning mode

Browse files
Files changed (2) hide show
  1. README.md +1 -0
  2. start.sh +17 -0
README.md CHANGED
@@ -30,6 +30,7 @@ Optional tuning variables:
30
  - `UBATCH_SIZE`: micro-batch size, default `512`
31
  - `CACHE_TYPE_K`: KV cache type for keys, default `q8_0`
32
  - `CACHE_TYPE_V`: KV cache type for values, default `q8_0`
 
33
  - `PARALLEL`: optional override for concurrent prompt decode slots
34
  - `PORT`: listen port, default `7860`
35
 
 
30
  - `UBATCH_SIZE`: micro-batch size, default `512`
31
  - `CACHE_TYPE_K`: KV cache type for keys, default `q8_0`
32
  - `CACHE_TYPE_V`: KV cache type for values, default `q8_0`
33
+ - `REASONING`: reasoning mode, accepts `True`, `False`, or `auto` and maps to llama.cpp `--reasoning`
34
  - `PARALLEL`: optional override for concurrent prompt decode slots
35
  - `PORT`: listen port, default `7860`
36
 
start.sh CHANGED
@@ -15,6 +15,7 @@ export QUANT_PREFERENCE="${QUANT_PREFERENCE:-q4_k_m,q4_k_s,q4_k,q4_0,q4_1}"
15
  export PERF_PROFILE="${PERF_PROFILE:-balanced}"
16
  export CACHE_TYPE_K="${CACHE_TYPE_K:-f16}"
17
  export CACHE_TYPE_V="${CACHE_TYPE_V:-f16}"
 
18
 
19
  # Ensure data directories exist and are writable. Run as root in container startup.
20
  echo "preparing storage: $HF_HOME and $MODEL_DIR"
@@ -115,6 +116,22 @@ server_args=(
115
  --metrics
116
  )
117
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
118
  if [[ -n "${PARALLEL:-}" ]]; then
119
  server_args+=(--parallel "$PARALLEL")
120
  fi
 
15
  export PERF_PROFILE="${PERF_PROFILE:-balanced}"
16
  export CACHE_TYPE_K="${CACHE_TYPE_K:-f16}"
17
  export CACHE_TYPE_V="${CACHE_TYPE_V:-f16}"
18
+ export REASONING="${REASONING:-auto}"
19
 
20
  # Ensure data directories exist and are writable. Run as root in container startup.
21
  echo "preparing storage: $HF_HOME and $MODEL_DIR"
 
116
  --metrics
117
  )
118
 
119
+ case "${REASONING,,}" in
120
+ true|1|on|yes)
121
+ server_args+=(--reasoning on)
122
+ ;;
123
+ false|0|off|no)
124
+ server_args+=(--reasoning off)
125
+ ;;
126
+ auto|"")
127
+ server_args+=(--reasoning auto)
128
+ ;;
129
+ *)
130
+ echo "invalid REASONING value: $REASONING (expected True/False/auto)"
131
+ exit 1
132
+ ;;
133
+ esac
134
+
135
  if [[ -n "${PARALLEL:-}" ]]; then
136
  server_args+=(--parallel "$PARALLEL")
137
  fi