雾聪
2023-12-05 2f5b2c57783e02a1a7771601367c1a36b8934ca6
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