diff --git a/README.md b/README.md index 71caa29..f2b261e 100644 --- a/README.md +++ b/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; diff --git a/README.zh-CN.md b/README.zh-CN.md index 5a73dc9..17aa3a1 100644 --- a/README.zh-CN.md +++ b/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; diff --git a/tests/messy_test.rs b/tests/messy_test.rs index de2dfa1..caa1257 100644 --- a/tests/messy_test.rs +++ b/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)?; diff --git a/todo.md b/todo.md index c8df364..6f47f23 100644 --- a/todo.md +++ b/todo.md @@ -6,3 +6,4 @@ * MinerU2.5-Pro-2604-1.2B * MiniCPM-V4.6 * Hy-MT1.5-1.8B-1.25bit +