stash save

2026-05-09 12:18:16 +08:00
parent 5df1f30f83
commit e8bf980b38
39 changed files with 2505 additions and 316 deletions
@@ -1,13 +1,5 @@
 use aha::models::{
-    deepseek_ocr::config::DeepseekOCRConfig,
-    hunyuan_ocr::config::HunYuanVLConfig,
-    lfm2::config::Lfm2Config,
-    lfm2vl::config::{Lfm2ProcessorConfig, Lfm2VLConfig},
-    minicpm4::config::MiniCPM4Config,
-    paddleocr_vl::config::PaddleOCRVLConfig,
-    qwen2_5vl::config::Qwen2_5VLConfig,
-    qwen3vl::config::Qwen3VLConfig,
-    voxcpm::config::VoxCPMConfig,
+    deepseek_ocr::config::DeepseekOCRConfig, hunyuan_ocr::config::HunYuanVLConfig, lfm2::config::Lfm2Config, lfm2vl::config::{Lfm2ProcessorConfig, Lfm2VLConfig}, minicpm4::config::MiniCPM4Config, moss::config::{MossAudioTokenizerConfig, MossTTSConfig}, paddleocr_vl::config::PaddleOCRVLConfig, qwen2_5vl::config::Qwen2_5VLConfig, qwen3vl::config::Qwen3VLConfig, voxcpm::config::VoxCPMConfig
 };
 use anyhow::Result;

@@ -117,3 +109,23 @@ fn lfm2vl_config() -> Result<()> {
    println!("{:?}", processor_config);
    Ok(())
 }
+
+#[test]
+fn moss_audio_tokenizer_config() -> Result<()> {
+    // cargo test -F cuda --test config_tests moss_audio_tokenizer_config -r -- --nocapture
+    let model_path = "/home/jhq/.aha/openmoss/MOSS-Audio-Tokenizer-Nano/";
+    let config_path = model_path.to_string() + "/config.json";
+    let config: MossAudioTokenizerConfig = serde_json::from_slice(&std::fs::read(config_path)?)?;
+    println!("{:?}", config);
+    Ok(())
+}
+
+#[test]
+fn moss_tts_config() -> Result<()> {
+    // cargo test -F cuda --test config_tests moss_tts_config -r -- --nocapture
+    let model_path = "/home/jhq/.aha/openmoss/MOSS-TTS-Nano/";
+    let config_path = model_path.to_string() + "/config.json";
+    let config: MossTTSConfig = serde_json::from_slice(&std::fs::read(config_path)?)?;
+    println!("{:?}", config);
+    Ok(())
+}