From a1cbcc09f4e99ffac0a7bee93775c9166cb66787 Mon Sep 17 00:00:00 2001
From: 游雁 <zhifu.gzf@alibaba-inc.com>
Date: 星期五, 12 五月 2023 10:43:03 +0800
Subject: [PATCH] onnx export funasr_onnx
---
funasr/datasets/large_datasets/utils/tokenize.py | 2 +-
1 files changed, 1 insertions(+), 1 deletions(-)
diff --git a/funasr/datasets/large_datasets/utils/tokenize.py b/funasr/datasets/large_datasets/utils/tokenize.py
index f8833b1..f0f0c66 100644
--- a/funasr/datasets/large_datasets/utils/tokenize.py
+++ b/funasr/datasets/large_datasets/utils/tokenize.py
@@ -57,7 +57,7 @@
length = len(text)
if 'hw_tag' in data:
hotword_indxs = sample_hotword(length, **hw_config)
- data[hotword_indxs] = hotword_indxs
+ data['hotword_indxs'] = hotword_indxs
del data['hw_tag']
for i in range(length):
x = text[i]
--
Gitblit v1.9.1