From 6e69d784e4814c3dbe35e8f70c6cf4b920c8b20b Mon Sep 17 00:00:00 2001
From: 天地 <tiandiweizun@gmail.com>
Date: 星期三, 19 三月 2025 23:10:13 +0800
Subject: [PATCH] 1. bug fix:list(mean)和list(var),由于mean和var是numpy,导致写入到文件的格式错误,参考上面的话,大概率是list(mean.tolist()),其实外层list没有必要 (#2437)

---
 funasr/utils/load_utils.py |    4 +++-
 1 files changed, 3 insertions(+), 1 deletions(-)

diff --git a/funasr/utils/load_utils.py b/funasr/utils/load_utils.py
index 1d80fcf..072b578 100644
--- a/funasr/utils/load_utils.py
+++ b/funasr/utils/load_utils.py
@@ -86,8 +86,10 @@
     ):  # download url to local file
         data_or_path_or_list = download_from_url(data_or_path_or_list)
 
-    if isinstance(data_or_path_or_list, str) and os.path.exists(data_or_path_or_list):  # local file
+    if (isinstance(data_or_path_or_list, str) and os.path.exists(data_or_path_or_list)) or hasattr(data_or_path_or_list, 'read'):  # local file or bytes io
         if data_type is None or data_type == "sound":
+            if hasattr(data_or_path_or_list, "read") and hasattr(data_or_path_or_list, "seek"):
+                data_or_path_or_list.seek(0)
             # if use_ffmpeg:
             #     data_or_path_or_list = _load_audio_ffmpeg(data_or_path_or_list, sr=fs)
             #     data_or_path_or_list = torch.from_numpy(data_or_path_or_list).squeeze()  # [n_samples,]

--
Gitblit v1.9.1