Skip to content

Instantly share code, notes, and snippets.

View baptistejamin's full-sized avatar

Baptiste Jamin baptistejamin

View GitHub Profile
#!/bin/bash
# Benchmark script to measure performance impact of logprobs with different topK settings
# Usage: ./scripts/benchmark_logprobs.sh [model_name]
set -e
MODEL="${1:-gemma3:1b}"
PROMPT="Write a short story about a robot."
NUM_PREDICT=100
RUNS=5