rust/sbv2-api
1
0
Fork 0
mirror of https://github.com/neodyland/sbv2-api.git synced 2026-05-29 20:20:36 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: tokenizer path, library, cuda, better performance

Browse Source
This commit is contained in:
Googlefan
2024-09-11 02:11:59 +00:00
parent 59efac6905
commit 20019a007c
16 changed files with 107 additions and 22191 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
sbv2_api/src/main.rs
View File

@@ -9,6 +9,7 @@ use sbv2_core::tts::TTSModel;
use serde::Deserialize;
use std::env;
use std::sync::Arc;
use tokio::fs;
use tokio::sync::Mutex;
mod error;
@@ -29,9 +30,10 @@ async fn synthesize(
tts_model
} else {
*tts_model = Some(TTSModel::new(
&env::var("BERT_MODEL_PATH")?,
&env::var("MAIN_MODEL_PATH")?,
&env::var("STYLE_VECTORS_PATH")?,
&fs::read(env::var("BERT_MODEL_PATH")?).await?,
&fs::read(env::var("MAIN_MODEL_PATH")?).await?,
&fs::read(env::var("STYLE_VECTORS_PATH")?).await?,
&fs::read(env::var("TOKENIZER_PATH")?).await?,
)?);
tts_model.as_ref().unwrap()
};