游雁
2024-02-05 08e6f946aa4c02e3ba3cebb47af2ac3cd5abe97b
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