update messy

This commit is contained in:
jhqxxx
2026-04-22 16:53:51 +08:00
parent dc15a48d14
commit 9e1adff2a1
8 changed files with 90 additions and 120 deletions
+15
View File
@@ -410,3 +410,18 @@ fn fire_red_vad_weight() -> Result<()> {
Ok(())
}
#[test]
fn silero_vad_weight() -> Result<()> {
// cargo test -F cuda --test weight_test silero_vad_weight -r -- --nocapture
let save_dir =
aha::utils::get_default_save_dir().ok_or(anyhow::anyhow!("Failed to get save dir"))?;
let model_path = format!("{}/silero_vad/silero_vad_16k.safetensors", save_dir);
let device = get_device(None);
let weights = safetensors::load(model_path, &device)?;
for (key, tensor) in weights.iter() {
println!("=== {} === {:?}", key, tensor);
}
Ok(())
}