423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2025-08-01 11:12:50 +08:00
Code Issues Packages Projects Releases Wiki Activity

[data] fix audio reader (#8448)

Browse Source
This commit is contained in:
Yaowei Zheng 2025-06-24 20:53:20 +08:00 committed by GitHub
parent 31b0787e12
commit be27eae175
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
src/llamafactory/data/mm_plugin.py
View File

@ -298,11 +298,8 @@ class MMPluginMixin:
r"""Regularizes audios to avoid error. Including reading and resampling.""" r"""Regularizes audios to avoid error. Including reading and resampling."""
results, sampling_rates = [], [] results, sampling_rates = [], []
for audio in audios: for audio in audios:
if isinstance(audio, (str, BinaryIO)):
audio, sampling_rate = librosa.load(audio, sr=sampling_rate)
if not isinstance(audio, np.ndarray): if not isinstance(audio, np.ndarray):
raise ValueError(f"Expect input is a list of audios, but got {type(audio)}.") audio, sampling_rate = librosa.load(audio, sr=sampling_rate)
results.append(audio) results.append(audio)
sampling_rates.append(sampling_rate) sampling_rates.append(sampling_rate)