refactor: reorganize imports and improve code formatting across multiple modules
This commit is contained in:
+7
-4
@@ -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()?;
|
||||
|
||||
@@ -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(())
|
||||
}
|
||||
}
|
||||
|
||||
@@ -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",
|
||||
|
||||
@@ -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(())
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user