gigastt-core 2.18.0

Core inference engine for gigastt — GigaAM v3 ONNX Runtime, model management, quantization
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60

//! Model download and management.
//!
//! Downloads GigaAM v3 RNN-T ONNX files from HuggingFace to `~/.gigastt/models/`.
//! Two recognition heads are selectable via [`ModelVariant`]: the plain `rnnt`
//! head (default — lower WER, bare lowercase output) and the `e2e_rnnt` head
//! (punctuation / casing / ITN baked in).
//!
//! An optional `manifest.toml` in the model directory can override ONNX/vocab
//! basenames and select the architecture for third-party packs (see
//! [`ModelManifest`]). When absent, load paths use the hardcoded
//! [`ModelVariant`] filenames.

mod cache;
mod download;
mod manifest;
mod progress;
mod variant;

#[cfg(feature = "coreml")]
pub(crate) use cache::coreml_cache_dir;
pub use cache::{
    CoremlCachePruneReport, DedupeReport, OptimizedCachePruneReport, dedupe_model_dir,
    optimized_cache_basename, prune_coreml_cache, prune_optimized_cache,
};
pub use manifest::{MANIFEST_FILE, ManifestFiles, ModelManifest};

#[cfg(all(feature = "net", feature = "ane"))]
pub use download::ensure_ane_packages;
#[cfg(all(feature = "net", feature = "diarization"))]
pub use download::ensure_speaker_model;
pub(crate) use download::verify_pinned_checksum;
pub use download::{
    VariantAction, default_model_dir, default_punct_model_dir, default_vad_model_dir,
    is_model_present, is_prequantized_present, is_usable_present, resolve_variant,
};
#[cfg(feature = "ane")]
pub use download::{
    ane_package_complete, ane_package_dir_name, default_ane_model_dir, is_ane_present,
};
#[cfg(feature = "net")]
pub use download::{
    ensure_fp32_model_variant, ensure_model, ensure_model_variant,
    ensure_prequantized_model_variant, ensure_punct_model, ensure_vad_model,
};

#[cfg(feature = "net")]
pub use progress::classify_download_error;
pub use progress::{
    ProgressErrorKind, ProgressEvent, ProgressMode, emit_progress_event, progress_mode,
    set_progress_mode,
};

#[cfg(feature = "ane")]
pub use variant::ANE_BUCKETS;
pub use variant::ModelVariant;
#[cfg(feature = "diarization")]
pub use variant::SPEAKER_MODEL_FILE;

#[cfg(test)]
mod tests;