any-tts 0.1.0

A Rust TTS library with Candle backends and runtime adapters for modern open TTS models
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
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80

//! Native OmniVoice backend.

use crate::config::{
    preferred_runtime_choice as preferred_model_runtime_choice,
    preferred_runtime_choices as preferred_model_runtime_choices, DType, RuntimeChoice,
};
use crate::device::DeviceSelection;
use crate::models::ModelType;

mod audio_tokenizer;
mod config;
pub mod model;

pub use model::OmniVoiceModel;

/// Recommended runtime choice for native OmniVoice on the current binary.
#[derive(Debug, Clone, Copy, PartialEq, Eq)]
pub struct OmniVoiceRuntimeChoice {
    pub device: DeviceSelection,
    pub dtype: DType,
}

impl From<RuntimeChoice> for OmniVoiceRuntimeChoice {
    fn from(choice: RuntimeChoice) -> Self {
        Self {
            device: choice.device,
            dtype: choice.dtype,
        }
    }
}

impl OmniVoiceRuntimeChoice {
    /// Combined backend label used by examples and benchmarks.
    pub fn label(&self) -> String {
        format!("{} ({})", self.device.label(), self.dtype.label())
    }

    /// Human-readable floating-point dtype label.
    pub fn dtype_label(&self) -> &'static str {
        self.dtype.label()
    }
}

/// Preferred OmniVoice runtime choices ordered by expected performance.
pub fn preferred_runtime_choices() -> Vec<OmniVoiceRuntimeChoice> {
    preferred_model_runtime_choices(ModelType::OmniVoice)
        .into_iter()
        .map(Into::into)
        .collect()
}

/// Best OmniVoice runtime choice for the current binary with CPU fallback.
pub fn preferred_runtime_choice() -> OmniVoiceRuntimeChoice {
    preferred_model_runtime_choice(ModelType::OmniVoice).into()
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn test_cpu_runtime_prefers_f32() {
        let choice = OmniVoiceRuntimeChoice {
            device: DeviceSelection::Cpu,
            dtype: DType::F32,
        };
        assert_eq!(choice.dtype, DType::F32);
        assert_eq!(choice.label(), "cpu (f32)");
    }

    #[test]
    fn test_metal_runtime_prefers_f32() {
        let choice = OmniVoiceRuntimeChoice {
            device: DeviceSelection::Metal(0),
            dtype: DType::F32,
        };
        assert_eq!(choice.dtype, DType::F32);
        assert_eq!(choice.label(), "metal:0 (f32)");
    }
}