rust/sbv2-api
1
0
Fork 0
mirror of https://github.com/neodyland/sbv2-api.git synced 2026-01-06 06:22:57 +00:00
Code Issues Packages Projects Releases Wiki Activity

split the code for support voicevox

Browse Source
This commit is contained in:
Masato Kikuchi
2025-03-26 15:14:22 +09:00
parent 99a4b130af
commit fc944b9d33
6 changed files with 45 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
crates/sbv2_core/src/tts_util.rs
View File

@@ -5,6 +5,19 @@ use crate::{jtalk, nlp, norm, tokenizer, utils};
use hound::{SampleFormat, WavSpec, WavWriter};
use ndarray::{concatenate, s, Array, Array1, Array2, Array3, Axis};
use tokenizers::Tokenizer;
pub fn preprocess_parse_text(
text: &str,
jtalk: &jtalk::JTalk,
) -> Result<(Vec<String>, Vec<i32>, Vec<i32>)> {
let text = jtalk.num2word(text)?;
let normalized_text = norm::normalize_text(&text);
let process = jtalk.process_text(&normalized_text)?;
let result = process.g2p()?;
Ok(result)
}
/// Parse text and return the input for synthesize
///
/// # Note
@@ -21,11 +34,7 @@ pub async fn parse_text(
Box<dyn std::future::Future<Output = Result<ndarray::Array2<f32>>>>,
>,
) -> Result<(Array2<f32>, Array1<i64>, Array1<i64>, Array1<i64>)> {
let text = jtalk.num2word(text)?;
let normalized_text = norm::normalize_text(&text);
let process = jtalk.process_text(&normalized_text)?;
let (phones, tones, mut word2ph) = process.g2p()?;
let (phones, tones, mut word2ph) = preprocess_parse_text(text, jtalk)?;
let (phones, tones, lang_ids) = nlp::cleaned_text_to_sequence(phones, tones);
let phones = utils::intersperse(&phones, 0);