bugfix for paraformer-streaming

This commit is contained in:
游雁 2024-08-02 01:55:30 +08:00
parent 7f6099140a
commit a28de72b17

View File

@ -76,7 +76,7 @@ def load_audio_text_image_video(
for audio in data_or_path_or_list
]
if isinstance(data_or_path_or_list, str) and data_or_path_or_list.startswith(
("http://", "https://")
("http://", "https://")
): # download url to local file
data_or_path_or_list = download_from_url(data_or_path_or_list)
@ -112,7 +112,7 @@ def load_audio_text_image_video(
elif isinstance(data_or_path_or_list, str) and data_type == "text" and tokenizer is not None:
data_or_path_or_list = tokenizer.encode(data_or_path_or_list)
elif isinstance(data_or_path_or_list, np.ndarray): # audio sample point
data_or_path_or_list = torch.from_numpy(data_or_path_or_list).squeeze() # [n_samples,]
data_or_path_or_list = torch.from_numpy(data_or_path_or_list) # .squeeze() # [n_samples,]
elif isinstance(data_or_path_or_list, str) and data_type == "kaldi_ark":
data_mat = kaldiio.load_mat(data_or_path_or_list)
if isinstance(data_mat, tuple):