From 68109d7458e04514b1b7e2c168c167f1bcfe7908 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E6=B8=B8=E9=9B=81?= Date: Thu, 23 Nov 2023 14:25:25 +0800 Subject: [PATCH] update funasr.text -> funasr.tokenizer fix bug export --- funasr/models/frontend/wav_frontend.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/funasr/models/frontend/wav_frontend.py b/funasr/models/frontend/wav_frontend.py index 2cf7ade7c..f92f32274 100644 --- a/funasr/models/frontend/wav_frontend.py +++ b/funasr/models/frontend/wav_frontend.py @@ -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,