shixian.shi
2023-06-27 a3c3e1a250072a54c12ef03d5cb7cfd51bdc6f05
update
1个文件已修改
5 ■■■■■ 已修改文件
funasr/datasets/large_datasets/utils/tokenize.py 5 ●●●●● 补丁 | 查看 | 原始文档 | blame | 历史
funasr/datasets/large_datasets/utils/tokenize.py
@@ -58,10 +58,11 @@
            # enable preset hotword detect in sampling
            pre_index = None
            for hw in hw_config['pre_hwlist']:
                _find = data['text'].find(hw)
                _find = data['text'][0].find(hw)
                if _find != -1:
                    _find = text[:_find].count(" ")  # bpe sometimes
                    _find = data['text'][0][:_find].count(" ")  # bpe sometimes
                    pre_index = [_find, _find + max(hw.count(" "), 1)]
                    import pdb; pdb.set_trace()
                    break
        hotword_indxs = sample_hotword(length, **hw_config, pre_index=pre_index)
        data['hotword_indxs'] = hotword_indxs