update funasr.text -> funasr.tokenizer fix bug export

This commit is contained in:
游雁 2023-11-23 14:25:25 +08:00
parent 61567c6d3b
commit 68109d7458

View File

@ -146,7 +146,7 @@ class WavFrontend(AbsFrontend):
feats_lens = torch.as_tensor(feats_lens)
if batch_size == 1:
feats_pad = feats[None, :, :]
feats_pad = feats[0][None, :, :]
else:
feats_pad = pad_sequence(feats,
batch_first=True,