语帆
2024-02-22 060d18ee4b86729e11f31ff16c822f3be33503ae
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