refactor: reorganize imports and improve code formatting across multiple modules

This commit is contained in:
XiaoYang
2026-02-08 11:07:53 +08:00
parent fcbe736e1a
commit 38ea72e407
24 changed files with 71 additions and 48 deletions
+7 -4
View File
@@ -2,9 +2,9 @@
use std::time::Instant;
use aha::utils::{tensor_utils::interpolate_nearest_1d};
use aha::utils::tensor_utils::interpolate_nearest_1d;
use anyhow::Result;
use candle_core::{Tensor};
use candle_core::Tensor;
// use symphonia::core::io::MediaSourceStream;
#[test]
@@ -16,8 +16,11 @@ fn messy_test() -> Result<()> {
let i_start = Instant::now();
let t_inter = interpolate_nearest_1d(&t, 20)?;
let i_duration = i_start.elapsed();
println!("Time elapsed in interpolate_nearest_1d is: {:?}", i_duration);
println!("t_inter: {}", t_inter);
println!(
"Time elapsed in interpolate_nearest_1d is: {:?}",
i_duration
);
println!("t_inter: {}", t_inter);
// let url = "https://sis-sample-audio.obs.cn-north-1.myhuaweicloud.com/16k16bit.mp3";
// let client = reqwest::blocking::Client::new();
// let response = client.get(url).send()?;
+2 -2
View File
@@ -1,8 +1,8 @@
use std::time::Instant;
use anyhow::Result;
use aha::models::index_tts2::{generate::IndexTTS2Generate, utils::download_index_tts2_need_model};
use aha_openai_dive::v1::resources::chat::ChatCompletionParameters;
use anyhow::Result;
#[tokio::test]
async fn index_tts2_generate() -> Result<()> {
@@ -45,4 +45,4 @@ async fn index_tts2_generate() -> Result<()> {
let i_duration = i_start.elapsed();
println!("Time elapsed in generate is: {:?}", i_duration);
Ok(())
}
}
+1 -1
View File
@@ -9,7 +9,7 @@ fn qwen3_asr_generate() -> Result<()> {
// RUST_BACKTRACE=1 cargo test -F cuda qwen3_asr_generate -r -- --nocapture
let save_dir =
aha::utils::get_default_save_dir().ok_or(anyhow::anyhow!("Failed to get save dir"))?;
let model_path = format!("{}/Qwen/Qwen3-ASR-0.6B/", save_dir); //Qwen/Qwen3-ASR-1.7B
let model_path = format!("{}/Qwen/Qwen3-ASR-0.6B/", save_dir); //Qwen/Qwen3-ASR-1.7B
let message = r#"
{
"model": "qwen3-asr",
+4 -4
View File
@@ -202,8 +202,8 @@ fn qwen3_weight() -> Result<()> {
fn index_tts2_weight() -> Result<()> {
let save_dir: String =
aha::utils::get_default_save_dir().ok_or(anyhow::anyhow!("Failed to get save dir"))?;
let model_path = format!("{}/IndexTeam/IndexTTS-2/", save_dir);
let s2mel_path = model_path+ "/s2mel.pth";
let model_path = format!("{}/IndexTeam/IndexTTS-2/", save_dir);
let s2mel_path = model_path + "/s2mel.pth";
// let wac2vec2_path = model_path+ "/wav2vec2bert_stats.pt";
// let model_path = format!("{}/iic/speech_campplus_sv_zh-cn_16k-common/", save_dir);
// let campplus_path = model_path+ "/campplus_cn_common.bin";
@@ -229,9 +229,9 @@ fn index_tts2_weight() -> Result<()> {
// let model_list = vec![semantic_codec_path];
// for m in model_list {
// let weights = safetensors::load(m, &device)?;
// for (key, tensor) in weights.iter() {
// for (key, tensor) in weights.iter() {
// println!("=== {} === {:?}", key, tensor.shape());
// }
// }
Ok(())
}
}