chore: fix autoresearch benchmark model id
This commit is contained in:
1 parent
f83e8b83f6
commit
b38c77f875
2 files changed
+24
-19
No files matched your search
@@ -6,6 +6,8 @@ ROOT="$SCRIPT_DIR"
|
|||||||
PORT="${LOCALLLM_BENCH_PORT:-18080}"
|
PORT="${LOCALLLM_BENCH_PORT:-18080}"
|
||||||
HOST="127.0.0.1:${PORT}"
|
HOST="127.0.0.1:${PORT}"
|
||||||
MODEL_PATH="${LOCALLLM_MODEL:-$ROOT/models/Qwen3.5-4B-Abliterated-Claude-4.6-Opus-Reasoning-Distilled.i1-Q6_K.gguf}"
|
MODEL_PATH="${LOCALLLM_MODEL:-$ROOT/models/Qwen3.5-4B-Abliterated-Claude-4.6-Opus-Reasoning-Distilled.i1-Q6_K.gguf}"
|
||||||
|
MODEL_ID="${LOCALLLM_BENCH_MODEL_ID:-$(basename "$MODEL_PATH")}"
|
||||||
|
MODEL_ID="${MODEL_ID%.gguf}"
|
||||||
SERVER_BIN="$ROOT/target/release/localllm.exe"
|
SERVER_BIN="$ROOT/target/release/localllm.exe"
|
||||||
LOG_DIR="$ROOT/logs"
|
LOG_DIR="$ROOT/logs"
|
||||||
SERVER_LOG="$LOG_DIR/autoresearch-server.log"
|
SERVER_LOG="$LOG_DIR/autoresearch-server.log"
|
||||||
@@ -39,4 +41,5 @@ SERVER_PID=$!
|
|||||||
|
|
||||||
python "$ROOT/experiments/bench_chat_latency.py" \
|
python "$ROOT/experiments/bench_chat_latency.py" \
|
||||||
--base-url "http://$HOST" \
|
--base-url "http://$HOST" \
|
||||||
|
--model-id "$MODEL_ID" \
|
||||||
--server-log "$SERVER_LOG"
|
--server-log "$SERVER_LOG"
|
||||||
@@ -9,21 +9,6 @@ import urllib.error
|
|||||||
import urllib.request
|
import urllib.request
|
||||||
|
|
||||||
|
|
||||||
PAYLOAD = {
|
|
||||||
"model": "local-model",
|
|
||||||
"stream": False,
|
|
||||||
"temperature": 0.0,
|
|
||||||
"top_p": 1.0,
|
|
||||||
"max_completion_tokens": 48,
|
|
||||||
"messages": [
|
|
||||||
{
|
|
||||||
"role": "user",
|
|
||||||
"content": "Write a compact Rust function that parses a comma-separated list of u32 values and returns a Vec<u32> while ignoring empty segments.",
|
|
||||||
}
|
|
||||||
],
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
def percentile(sorted_values, pct):
|
def percentile(sorted_values, pct):
|
||||||
if not sorted_values:
|
if not sorted_values:
|
||||||
return 0.0
|
return 0.0
|
||||||
@@ -63,8 +48,24 @@ def wait_for_health(base_url, timeout_s, server_log):
|
|||||||
sys.exit(1)
|
sys.exit(1)
|
||||||
|
|
||||||
|
|
||||||
def run_once(base_url):
|
def build_payload(model_id):
|
||||||
body = json.dumps(PAYLOAD).encode("utf-8")
|
return {
|
||||||
|
"model": model_id,
|
||||||
|
"stream": False,
|
||||||
|
"temperature": 0.0,
|
||||||
|
"top_p": 1.0,
|
||||||
|
"max_completion_tokens": 48,
|
||||||
|
"messages": [
|
||||||
|
{
|
||||||
|
"role": "user",
|
||||||
|
"content": "Write a compact Rust function that parses a comma-separated list of u32 values and returns a Vec<u32> while ignoring empty segments.",
|
||||||
|
}
|
||||||
|
],
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
def run_once(base_url, model_id):
|
||||||
|
body = json.dumps(build_payload(model_id)).encode("utf-8")
|
||||||
req = urllib.request.Request(
|
req = urllib.request.Request(
|
||||||
base_url + "/v1/chat/completions",
|
base_url + "/v1/chat/completions",
|
||||||
data=body,
|
data=body,
|
||||||
@@ -82,6 +83,7 @@ def run_once(base_url):
|
|||||||
def main():
|
def main():
|
||||||
parser = argparse.ArgumentParser()
|
parser = argparse.ArgumentParser()
|
||||||
parser.add_argument("--base-url", required=True)
|
parser.add_argument("--base-url", required=True)
|
||||||
|
parser.add_argument("--model-id", required=True)
|
||||||
parser.add_argument("--server-log", default="")
|
parser.add_argument("--server-log", default="")
|
||||||
parser.add_argument("--warmups", type=int, default=1)
|
parser.add_argument("--warmups", type=int, default=1)
|
||||||
parser.add_argument("--samples", type=int, default=7)
|
parser.add_argument("--samples", type=int, default=7)
|
||||||
@@ -91,12 +93,12 @@ def main():
|
|||||||
wait_for_health(args.base_url, args.health_timeout, args.server_log)
|
wait_for_health(args.base_url, args.health_timeout, args.server_log)
|
||||||
|
|
||||||
for _ in range(args.warmups):
|
for _ in range(args.warmups):
|
||||||
run_once(args.base_url)
|
run_once(args.base_url, args.model_id)
|
||||||
|
|
||||||
latencies = []
|
latencies = []
|
||||||
tps_values = []
|
tps_values = []
|
||||||
for _ in range(args.samples):
|
for _ in range(args.samples):
|
||||||
latency_ms, tps = run_once(args.base_url)
|
latency_ms, tps = run_once(args.base_url, args.model_id)
|
||||||
latencies.append(latency_ms)
|
latencies.append(latency_ms)
|
||||||
tps_values.append(tps)
|
tps_values.append(tps)
|
||||||
|
|
||||||
|
|||||||
Reference in new issue
Block a user