雾聪
2024-03-03 26f70e6b328e8df32e2ac1d8ba3a38c7cf7bdf29
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