speech_recognition/FunASR
1
0
Fork 0
mirror of https://github.com/modelscope/FunASR synced 2025-09-15 14:48:36 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

update funasr-onnx

Browse Source
This commit is contained in:
shixian.shi 2024-01-10 11:23:41 +08:00
parent f79d31d860
commit e30a17cf4e
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
runtime/python/onnxruntime/funasr_onnx/utils/utils.py
View File

@ -28,7 +28,7 @@ def pad_list(xs, pad_value, max_len=None):
max_len = max(x.size(0) for x in xs)
# pad = xs[0].new(n_batch, max_len, *xs[0].size()[1:]).fill_(pad_value)
# numpy format
pad = np.zeros((n_batch, max_len)).astype(np.int32)
pad = (np.zeros((n_batch, max_len)) + pad_value).astype(np.int32)
for i in range(n_batch):
pad[i, : xs[i].shape[0]] = xs[i]