Yabin Li
2023-12-18 a0108271ad3a5efeaff0054d7ea3625d4461178b
runtime/python/onnxruntime/funasr_onnx/punc_bin.py
@@ -276,7 +276,7 @@
        param_dict[cache_key] = cache_out
        return sentence_out, sentence_punc_list_out, cache_out
    def vad_mask(self, size, vad_pos, dtype=np.bool):
    def vad_mask(self, size, vad_pos, dtype=bool):
        """Create mask for decoder self-attention.
        :param int size: size of mask