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

feat: sbv2file format

Browse Source
This commit is contained in:
Googlefan
2024-09-11 11:20:25 +00:00
parent d4e2fe30d0
commit fd1caf8ab5
7 changed files with 117 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
sbv2_api/src/main.rs
View File

@@ -88,6 +88,20 @@ impl AppState {
.iter()
.collect::<String>(),
);
} else if name.ends_with(".sbv2") {
let entry = &name[..name.len() - 5];
log::info!("Try loading: {entry}");
let sbv2_bytes = match fs::read(format!("{models}/{entry}.sbv2")).await {
Ok(b) => b,
Err(e) => {
log::warn!("Error loading sbv2_bytes from file {entry}: {e}");
continue;
}
};
if let Err(e) = tts_model.load_sbv2file(&entry, sbv2_bytes) {
log::warn!("Error loading {entry}: {e}");
};
log::info!("Loaded: {entry}");
}
}
for entry in entries {
@@ -110,6 +124,7 @@ impl AppState {
if let Err(e) = tts_model.load(&entry, style_vectors_bytes, vits2_bytes) {
log::warn!("Error loading {entry}: {e}");
};
log::info!("Loaded: {entry}");
}
Ok(Self {
tts_model: Arc::new(Mutex::new(tts_model)),