update doc

README.md

@@ -216,7 +216,7 @@ pnpm run tauri build
 > cargo add aha
 
 ```rust
-# VoxCPM example
+// VoxCPM example
 use aha::models::voxcpm::generate::VoxCPMGenerate;
 use aha::utils::audio_utils::save_wav_mono;
 use anyhow::Result;

README.zh-CN.md

@@ -215,7 +215,7 @@ pnpm run tauri build
 > cargo add aha
 
 ```rust
-# VoxCPM示例
+// VoxCPM示例
 use aha::models::voxcpm::generate::VoxCPMGenerate;
 use aha::utils::audio_utils::save_wav_mono;
 use anyhow::Result;

tests/messy_test.rs

@@ -41,10 +41,12 @@ async fn download_test() -> Result<()> {
 #[test]
 fn messy_test() -> Result<()> {
     // RUST_BACKTRACE=1 cargo test -F cuda --test messy_test messy_test -r -- --nocapture
-    let device = aha::Device::Cpu;
-    let input = Tensor::new(&[5u32, 4, 3, 6], &device)?;
-    let mask = get_mask_from_lengths(&input)?;
-    println!("{}", mask);
+    // let device = aha::Device::Cpu;
+    let device = aha::Device::new_cuda(0)?;
+    println!("{:?}", device);
+    // let input = Tensor::new(&[5u32, 4, 3, 6], &device)?;
+    // let mask = get_mask_from_lengths(&input)?;
+    // println!("{}", mask);
     // let audio_path = "file:///home/jhq/python_code/FireRedASR2S/assets/hello_zh.wav";
     // let device = aha::Device::Cpu;
     // let wave = load_audio_with_resample(audio_path, &device, Some(16000), true)?;

todo.md

@@ -6,3 +6,4 @@
 * MinerU2.5-Pro-2604-1.2B
 * MiniCPM-V4.6
 * Hy-MT1.5-1.8B-1.25bit
+