From 4f98546f3693482f8f34aa5f11ced31381c58724 Mon Sep 17 00:00:00 2001
From: 游雁 <zhifu.gzf@alibaba-inc.com>
Date: 星期五, 05 一月 2024 16:55:07 +0800
Subject: [PATCH] load_audio_text_image_video

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

diff --git a/funasr/utils/load_utils.py b/funasr/utils/load_utils.py
index 7f1b850..637e1d2 100644
--- a/funasr/utils/load_utils.py
+++ b/funasr/utils/load_utils.py
@@ -27,7 +27,7 @@
 # 	return audio_or_path_or_list
 
 
-def load_audio_and_text_image_video(audio_or_path_or_list, fs: int = 16000, audio_fs: int = 16000, data_type=None, tokenizer=None):
+def load_audio_text_image_video(audio_or_path_or_list, fs: int = 16000, audio_fs: int = 16000, data_type=None, tokenizer=None):
 	if isinstance(audio_or_path_or_list, (list, tuple)):
 		if data_type is not None and isinstance(data_type, (list, tuple)):
 
@@ -37,12 +37,12 @@
 				
 				for j, (data_type_j, audio_or_path_or_list_j) in enumerate(zip(data_type_i, audio_or_path_or_list_i)):
 					
-					audio_or_path_or_list_j = load_audio_and_text_image_video(audio_or_path_or_list_j, fs=fs, audio_fs=audio_fs, data_type=data_type_j, tokenizer=tokenizer)
+					audio_or_path_or_list_j = load_audio_text_image_video(audio_or_path_or_list_j, fs=fs, audio_fs=audio_fs, data_type=data_type_j, tokenizer=tokenizer)
 					audio_or_path_or_list_ret[j].append(audio_or_path_or_list_j)
 
 			return audio_or_path_or_list_ret
 		else:
-			return [load_audio_and_text_image_video(audio, fs=fs, audio_fs=audio_fs) for audio in audio_or_path_or_list]
+			return [load_audio_text_image_video(audio, fs=fs, audio_fs=audio_fs) for audio in audio_or_path_or_list]
 	
 	if isinstance(audio_or_path_or_list, str) and os.path.exists(audio_or_path_or_list):
 		audio_or_path_or_list, audio_fs = torchaudio.load(audio_or_path_or_list)

--
Gitblit v1.9.1