rlx-tiny-tts 0.2.9

TinyTTS English text-to-speech (VITS2/MeloTTS, 44.1 kHz) for RLX — all backends
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
81
82
83

//! TinyTTS bundle configuration (`config.json`).

use std::collections::BTreeMap;

use anyhow::Result;
use serde::Deserialize;

#[derive(Debug, Clone, Deserialize)]
pub struct BundleConfig {
    pub model: String,
    #[serde(default = "default_sample_rate")]
    pub sample_rate: u32,
    #[serde(default = "default_true")]
    pub add_blank: bool,
    #[serde(default = "default_language")]
    pub language: String,
    #[serde(default)]
    pub speakers: BTreeMap<String, i64>,
    #[serde(default)]
    pub default_speaker: Option<String>,
    #[serde(default = "default_noise_scale")]
    pub noise_scale: f32,
    #[serde(default = "default_noise_scale_w")]
    pub noise_scale_w: f32,
    #[serde(default = "default_length_scale")]
    pub length_scale: f32,
    /// Latent channel width of the VITS prior / flow / decoder input (ONNX `z`).
    #[serde(default = "default_inter_channels")]
    pub inter_channels: usize,
    /// Speaker-conditioning channel width (ONNX `g`).
    #[serde(default = "default_gin_channels")]
    pub gin_channels: usize,
}

fn default_sample_rate() -> u32 {
    44100
}
fn default_true() -> bool {
    true
}
fn default_language() -> String {
    "EN".to_string()
}
fn default_noise_scale() -> f32 {
    0.667
}
fn default_noise_scale_w() -> f32 {
    0.8
}
fn default_length_scale() -> f32 {
    1.0
}
fn default_inter_channels() -> usize {
    80
}
fn default_gin_channels() -> usize {
    80
}

impl BundleConfig {
    pub fn from_json(s: &str) -> Result<Self> {
        Ok(serde_json::from_str(s)?)
    }

    /// Default speaker id (`default_speaker` name → id, else first, else 0).
    pub fn default_speaker(&self) -> i64 {
        if let Some(name) = &self.default_speaker {
            if let Some(id) = self.speakers.get(name) {
                return *id;
            }
        }
        self.speakers.values().next().copied().unwrap_or(0)
    }

    /// Resolve a speaker name (case-insensitive) to its id, falling back to default.
    pub fn speaker_id(&self, name: &str) -> i64 {
        self.speakers
            .iter()
            .find(|(k, _)| k.eq_ignore_ascii_case(name))
            .map(|(_, v)| *v)
            .unwrap_or_else(|| self.default_speaker())
    }
}