Просмотр исходного кода

Merge branch 'main' of github.com:alibaba-damo-academy/FunASR
add

游雁 2 лет назад
Родитель
Сommit
62da61f857
1 измененных файлов с 1 добавлено и 1 удалено
  1. 1 1
      runtime/python/onnxruntime/funasr_onnx/punc_bin.py

+ 1 - 1
runtime/python/onnxruntime/funasr_onnx/punc_bin.py

@@ -276,7 +276,7 @@ class CT_Transformer_VadRealtime(CT_Transformer):
         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