stash save

tests/test_moss_tts.rs

@@ -0,0 +1,20 @@
+use aha::models::moss::generate::MossTTSGenerate;
+use anyhow::Result;
+
+#[test]
+fn moss_tts() -> Result<()> {
+    // RUST_BACKTRACE=1 cargo test -F cuda --test test_moss_tts moss_tts -r -- --nocapture
+    let save_dir =
+        aha::utils::get_default_save_dir().ok_or(anyhow::anyhow!("Failed to get save dir"))?;
+    let tts_path = format!("{}/openmoss/MOSS-TTS-Nano/", save_dir);
+    let audio_tokenizer_path = format!("{}/openmoss/MOSS-Audio-Tokenizer-Nano/", save_dir);
+    let mut model = MossTTSGenerate::init(&tts_path, &audio_tokenizer_path, None, None)?;
+    let _ = model.generate(
+        "您好啊,吃饭了吗,吃的啥啊中午",
+        Some("file://./assets/audio/jiangjiang.wav"),
+        Some("哈喽大家好，我是蒋蒋"),
+        Some(aha::models::moss::tts_nano::MossTTSMode::Continuation),
+        // None,
+    )?;
+    Ok(())
+}