K3-Stuff / scripts /26_draft_sweep.sh
TessaCoil's picture
Upload folder using huggingface_hub
ddf8c5b verified
Raw
History Blame Contribute Delete
1.07 kB
#!/usr/bin/env bash
# k3-test/26_draft_sweep.sh — sweep DSpark --draft-max; one server per depth (flag is load-time).
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
export K3TEST_ROOT="$SCRIPT_DIR"
source "$SCRIPT_DIR/lib/common.sh"
source "$K3TEST_ROOT/lib/server.sh"
set -uo pipefail
[[ -f $K3_MODEL_PATH ]] || die "K3 missing"
CODING="$K3TEST_ROOT/prompts/coding.txt"
DRAFT="$MODEL_DIR/k3-draft/draft.gguf"
[[ -f $DRAFT ]] || die "draft missing"
THREADS="${THREADS:-56}"
BASE_SPLIT="0.3,1,1,1,1,1,1,1"
BASE=( -m "$K3_MODEL_PATH" -ngl 999 --tensor-split "$BASE_SPLIT" --cpu-moe -b 512 -ub 512 )
for MAX in 4 7 10 14; do
if start_server "Dmax${MAX}" "${BASE[@]}" --spec-type draft-dspark -md "$DRAFT" --draft-max "$MAX" -np 1 -t "$THREADS"; then
server_completion "k3_D_max${MAX}_cold" "$CODING" 128
server_completion "k3_D_max${MAX}_warm" "$CODING" 128
stop_server
else
echo "Dmax${MAX} failed to load" | tee "$LOG_ROOT/k3_D_max${MAX}_FAIL.txt"
fi
done
echo "$(date +%s) 26_draft_sweep done" >> "$LOG_ROOT/k3_suite_status.txt"