Add REASONING env toggle for llama-server reasoning mode
Browse files
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
|