From f2a208d0d2e6a7805c44cd83faa7c313e01e594c Mon Sep 17 00:00:00 2001
From: SWHL <liekkaskono@163.com>
Date: 星期二, 21 二月 2023 15:38:38 +0800
Subject: [PATCH] Add code of generating whl about rapid_paraformer
---
funasr/export/models/e2e_asr_paraformer.py | 2 +-
1 files changed, 1 insertions(+), 1 deletions(-)
diff --git a/funasr/export/models/e2e_asr_paraformer.py b/funasr/export/models/e2e_asr_paraformer.py
index 84dd9d2..bf5ed1e 100644
--- a/funasr/export/models/e2e_asr_paraformer.py
+++ b/funasr/export/models/e2e_asr_paraformer.py
@@ -59,7 +59,7 @@
enc, enc_len = self.encoder(**batch)
mask = self.make_pad_mask(enc_len)[:, None, :]
pre_acoustic_embeds, pre_token_length, alphas, pre_peak_index = self.predictor(enc, mask)
- pre_token_length = pre_token_length.round().long()
+ pre_token_length = pre_token_length.round().type(torch.int32)
decoder_out, _ = self.decoder(enc, enc_len, pre_acoustic_embeds, pre_token_length)
decoder_out = torch.log_softmax(decoder_out, dim=-1)
--
Gitblit v1.9.1