嘉渊
2023-07-21 d273f7e12693e5b366cbf2ff7d01dde0264b01d9
funasr/datasets/large_datasets/utils/tokenize.py
@@ -59,11 +59,10 @@
            pre_index = None
            for hw in hw_config['pre_hwlist']:
                hw = " ".join(seg_tokenize(hw, seg_dict))
                _find = text.find(hw)
                _find = " ".join(text).find(hw)
                if _find != -1:
                    _find = text[:_find].count(" ")  # bpe sometimes
                    # _find = text[:_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