[package]
edition = "2024"
rust-version = "1.95"
name = "whisperforge-core"
version = "0.5.5"
authors = ["bevsxyz"]
build = false
autolib = false
autobins = false
autoexamples = false
autotests = false
autobenches = false
description = "GPU-accelerated Whisper model inference with streaming audio, quantization, and KV-cached decoding"
homepage = "https://github.com/bevsxyz/WhisperForge"
readme = "README.md"
keywords = [
"whisper",
"speech-to-text",
"gpu",
"quantization",
"burn",
]
categories = [
"multimedia::audio",
"science",
]
license = "MIT"
repository = "https://github.com/bevsxyz/WhisperForge"
resolver = "2"
[features]
cubecl-stft = [
"dep:cubecl",
"dep:burn-wgpu",
]
default = ["file-io"]
file-io = ["dep:symphonia"]
[lib]
name = "whisperforge_core"
path = "src/lib.rs"
[[test]]
name = "model_shapes"
path = "tests/model_shapes.rs"
[[test]]
name = "stream_integration"
path = "tests/stream_integration.rs"
[[test]]
name = "wer_benchmark"
path = "tests/wer_benchmark.rs"
[dependencies.anyhow]
version = "1.0"
[dependencies.audioadapter-buffers]
version = "3.0"
[dependencies.burn]
version = "0.21.0"
features = [
"std",
"burn-flex",
"std",
"burn-flex",
]
[dependencies.burn-flex]
version = "0.21.0"
[dependencies.burn-wgpu]
version = "0.21.0"
optional = true
[dependencies.cpal]
version = "0.15"
[dependencies.cubecl]
version = "0.10.0"
optional = true
[dependencies.flate2]
version = "1.0"
[dependencies.hf-hub]
version = "0.5"
features = ["ureq"]
default-features = false
[dependencies.hound]
version = "3.5"
[dependencies.ort]
version = "=2.0.0-rc.10"
features = [
"std",
"download-binaries",
"copy-dylibs",
]
default-features = false
[dependencies.rand]
version = "0.10"
[dependencies.ringbuf]
version = "0.4"
[dependencies.rubato]
version = "3.0"
[dependencies.rustfft]
version = "6.2"
[dependencies.serde]
version = "1.0"
features = ["derive"]
[dependencies.serde_json]
version = "1.0"
[dependencies.symphonia]
version = "0.6"
features = [
"pcm",
"wav",
"flac",
"vorbis",
"ogg",
"mkv",
"mp3",
"aac",
"isomp4",
]
optional = true
default-features = false
[dependencies.tokenizers]
version = "0.23"
features = ["onig"]
default-features = false
[dependencies.tracing]
version = "0.1"
[lints.clippy]
legacy_numeric_constants = "allow"
let_and_return = "allow"
single_range_in_vec_init = "allow"