From 7012ca2efc130103c4acd24e3678c7ae280f8db4 Mon Sep 17 00:00:00 2001
From: 游雁 <zhifu.gzf@alibaba-inc.com>
Date: 星期三, 13 十二月 2023 20:08:55 +0800
Subject: [PATCH] funasr2 paraformer biciparaformer contextuaparaformer
---
funasr/datasets/small_datasets/preprocessor.py | 1 +
1 files changed, 1 insertions(+), 0 deletions(-)
diff --git a/funasr/datasets/small_datasets/preprocessor.py b/funasr/datasets/small_datasets/preprocessor.py
index 01a8c6f..62beaab 100644
--- a/funasr/datasets/small_datasets/preprocessor.py
+++ b/funasr/datasets/small_datasets/preprocessor.py
@@ -361,6 +361,7 @@
tokens = seg_tokenize(tokens, self.seg_dict)
else:
tokens = self.tokenizer.text2tokens(text)
+
text_ints = self.token_id_converter.tokens2ids(tokens)
data[self.text_name] = np.array(text_ints, dtype=np.int64)
return data
--
Gitblit v1.9.1