DFlash causes premature EOS on a generating slot when another task lands with high LCP prompt reuse (shared draft ctx corruption, reproduced 2026-08-14 3x). Benching ngram-mod vs MTP sidecar to pick the replacement. serve script now passes --spec-draft-model only for draft-* spec types.
93 lines
2.7 KiB
Bash
Executable File
93 lines
2.7 KiB
Bash
Executable File
#!/usr/bin/env bash
|
|
# Serve HauhauCS Qwen3.6-27B-Uncensored-HauhauCS-Aggressive:Q8_K_P
|
|
# with DFlash draft speculative decoding (draft-dflash) on dankclaw
|
|
# (AMD Strix Halo gfx1151, ROCm backend).
|
|
# Runtime: llama.cpp llama-server.
|
|
#
|
|
# Tunables come from frank-default-large.env in this directory unless the
|
|
# FRANK_LARGE_ENV_FILE environment variable overrides that path.
|
|
set -euo pipefail
|
|
|
|
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
|
ENV_FILE="${FRANK_LARGE_ENV_FILE:-$SCRIPT_DIR/frank-default-large.env}"
|
|
if [[ -f "$ENV_FILE" ]]; then
|
|
set -a
|
|
# shellcheck disable=SC1090
|
|
source "$ENV_FILE"
|
|
set +a
|
|
fi
|
|
|
|
# --- defaults (used when the env file does not set them) -----------------
|
|
: "${MODEL_DIR:=/home/dank/models}"
|
|
: "${MODEL_FILE:=Qwen3.6-27B-Uncensored-HauhauCS-Aggressive-Q8_K_P.gguf}"
|
|
: "${DRAFT_FILE:=Qwen3.6-27B-DFlash-Q8_0.gguf}"
|
|
: "${ALIAS:=frank-default-large}"
|
|
: "${HOST:=0.0.0.0}"
|
|
: "${PORT:=7770}"
|
|
: "${CTX_SIZE:=602100}"
|
|
: "${PARALLEL:=3}"
|
|
: "${SPEC_TYPE:=draft-dflash}"
|
|
: "${SPEC_DRAFT_N_MAX:=8}"
|
|
: "${SPEC_NGRAM_MOD_N_MIN:=48}"
|
|
: "${SPEC_NGRAM_MOD_N_MAX:=64}"
|
|
: "${SPEC_NGRAM_MOD_N_MATCH:=24}"
|
|
: "${DEVICE:=ROCm0}"
|
|
: "${N_GPU_LAYERS:=all}"
|
|
: "${CACHE_TYPE_K:=q8_0}"
|
|
: "${CACHE_TYPE_V:=q8_0}"
|
|
: "${BATCH_SIZE:=8192}"
|
|
: "${UBATCH_SIZE:=2048}"
|
|
: "${TEMPERATURE:=1.0}"
|
|
: "${TOP_K:=20}"
|
|
: "${TOP_P:=0.95}"
|
|
: "${MIN_P:=0.0}"
|
|
: "${THREADS:=12}"
|
|
: "${THREADS_HTTP:=1}"
|
|
|
|
LLAMA_SERVER="${LLAMA_SERVER:-/home/dank/llama.cpp/build/bin/llama-server}"
|
|
MODEL="$MODEL_DIR/$MODEL_FILE"
|
|
DRAFT="$MODEL_DIR/$DRAFT_FILE"
|
|
|
|
if [[ ! -f "$MODEL" ]]; then
|
|
echo "missing required file: $MODEL" >&2
|
|
exit 1
|
|
fi
|
|
|
|
# Draft-model flags apply only to spec types that use a draft file
|
|
# (draft-dflash, draft-mtp, draft-eagle3). ngram-mod is draftless.
|
|
SPEC_ARGS=()
|
|
if [[ "$SPEC_TYPE" == *draft-* ]]; then
|
|
if [[ ! -f "$DRAFT" ]]; then
|
|
echo "missing required file: $DRAFT" >&2
|
|
exit 1
|
|
fi
|
|
SPEC_ARGS+=(--spec-draft-model "$DRAFT")
|
|
fi
|
|
|
|
exec "$LLAMA_SERVER" \
|
|
--model "$MODEL" \
|
|
"${SPEC_ARGS[@]}" \
|
|
--spec-type "$SPEC_TYPE" \
|
|
--spec-draft-n-max "$SPEC_DRAFT_N_MAX" \
|
|
--spec-ngram-mod-n-min "$SPEC_NGRAM_MOD_N_MIN" \
|
|
--spec-ngram-mod-n-max "$SPEC_NGRAM_MOD_N_MAX" \
|
|
--spec-ngram-mod-n-match "$SPEC_NGRAM_MOD_N_MATCH" \
|
|
--device "$DEVICE" \
|
|
--gpu-layers "$N_GPU_LAYERS" \
|
|
--fit off \
|
|
--cache-type-k "$CACHE_TYPE_K" \
|
|
--cache-type-v "$CACHE_TYPE_V" \
|
|
--ctx-size "$CTX_SIZE" \
|
|
--parallel "$PARALLEL" \
|
|
--batch-size "$BATCH_SIZE" \
|
|
--ubatch-size "$UBATCH_SIZE" \
|
|
--temp "$TEMPERATURE" \
|
|
--top-k "$TOP_K" \
|
|
--top-p "$TOP_P" \
|
|
--min-p "$MIN_P" \
|
|
--threads "$THREADS" \
|
|
--threads-http "$THREADS_HTTP" \
|
|
--alias "$ALIAS" \
|
|
--host "$HOST" \
|
|
--port "$PORT"
|