Merge branch 'pr/KingBright/13'

2026-01-08 00:12:23 +08:00
parent 618d75b263 4e35866a59
commit d5ec315150
1 changed files with 4 additions and 2 deletions
@@ -134,8 +134,10 @@ fn hunyuanocr_weight() -> Result<()> {

 #[test]
 fn glm_asr_nano_weight() -> Result<()> {
-    let model_path = "/home/jhq/huggingface_model/ZhipuAI/GLM-ASR-Nano-2512/";
-    let model_list = find_type_files(model_path, "safetensors")?;
+    let save_dir =
+        aha::utils::get_default_save_dir().ok_or(anyhow::anyhow!("Failed to get save dir"))?;
+    let model_path = format!("{}/ZhipuAI/GLM-ASR-Nano-2512/", save_dir);
+    let model_list = find_type_files(&model_path, "safetensors")?;

    let device = Device::Cpu;
    for m in &model_list {