Parakeet v Whisper h2h #26

Merged
iamdoubz merged 21 commits from test_parakeet into main 2026-07-16 10:23:17 -05:00
Showing only changes of commit 616d2f4f97 - Show all commits
+37 -12
View File
@@ -84,7 +84,13 @@ mod cputime {
fn normalize(s: &str) -> Vec<String> {
s.to_lowercase()
.chars()
.map(|c| if c.is_alphanumeric() || c == '\'' { c } else { ' ' })
.map(|c| {
if c.is_alphanumeric() || c == '\'' {
c
} else {
' '
}
})
.collect::<String>()
.split_whitespace()
.map(str::to_string)
@@ -121,6 +127,8 @@ fn self_test() {
// ---- engines ----
// ponytail: one Engine per process for its whole lifetime — variant size skew is irrelevant.
#[allow(clippy::large_enum_variant)]
enum Engine {
Whisper(WhisperTranscriber),
Onnx(OnnxTranscriber),
@@ -282,12 +290,15 @@ fn run_corpus(contender: &str, engine: &Engine, load_ms: u128, corpus: &Path, ou
(0usize, 0usize, 0u128, 0u64, 0f64);
for wav in &wavs {
let stem = wav.file_stem().unwrap().to_string_lossy().into_owned();
let reference =
std::fs::read_to_string(wav.with_extension("txt")).unwrap_or_else(|e| {
panic!("missing reference {}: {e}", wav.with_extension("txt").display())
});
let audio_secs =
read_wav_mono_16k(wav).map(|s| s.len() as f64 / SAMPLE_RATE as f64).unwrap_or(0.0);
let reference = std::fs::read_to_string(wav.with_extension("txt")).unwrap_or_else(|e| {
panic!(
"missing reference {}: {e}",
wav.with_extension("txt").display()
)
});
let audio_secs = read_wav_mono_16k(wav)
.map(|s| s.len() as f64 / SAMPLE_RATE as f64)
.unwrap_or(0.0);
let cpu0 = cputime::process_cpu_ms();
let t0 = Instant::now();
@@ -324,9 +335,18 @@ fn run_corpus(contender: &str, engine: &Engine, load_ms: u128, corpus: &Path, ou
}
println!("---- {contender} summary ----");
println!("files: {} audio: {audio_total:.1}s load: {load_ms}ms", wavs.len());
println!("WER: {:.2}% ({errors}/{words})", 100.0 * errors as f64 / words.max(1) as f64);
println!("RTF (wall): {:.3}", wall_total as f64 / 1000.0 / audio_total.max(0.001));
println!(
"files: {} audio: {audio_total:.1}s load: {load_ms}ms",
wavs.len()
);
println!(
"WER: {:.2}% ({errors}/{words})",
100.0 * errors as f64 / words.max(1) as f64
);
println!(
"RTF (wall): {:.3}",
wall_total as f64 / 1000.0 / audio_total.max(0.001)
);
println!(
"CPU-sec per audio-sec: {:.3}",
cpu_total as f64 / 1000.0 / audio_total.max(0.001)
@@ -349,7 +369,10 @@ fn run_live(contender: &str, engine: &Engine, wav: &Path) {
}
latencies.sort_unstable();
let pct = |p: f64| latencies[(((latencies.len() - 1) as f64) * p) as usize];
println!("---- {contender} live ({WINDOW_SECS}s windows, n={}) ----", latencies.len());
println!(
"---- {contender} live ({WINDOW_SECS}s windows, n={}) ----",
latencies.len()
);
println!(
"p50={}ms p95={}ms max={}ms (budget: {}ms)",
pct(0.50),
@@ -365,7 +388,9 @@ fn main() {
self_test();
return;
}
let contender = args.first().expect("usage: asr_bench <contender> [--corpus DIR|--live WAV]");
let contender = args
.first()
.expect("usage: asr_bench <contender> [--corpus DIR|--live WAV]");
let flag = |name: &str| {
args.iter()
.position(|a| a == name)