speech_asr
2023-03-15 2f933cb101e56c3c12c76c38d368b94111b52f64
update
2个文件已修改
6 ■■■■■ 已修改文件
funasr/bin/eend_ola_inference.py 4 ●●●● 补丁 | 查看 | 原始文档 | blame | 历史
funasr/models/e2e_diar_eend_ola.py 2 ●●●●● 补丁 | 查看 | 原始文档 | blame | 历史
funasr/bin/eend_ola_inference.py
@@ -27,9 +27,6 @@
from funasr.utils.types import str2triple_str
from funasr.utils.types import str_or_none
from modelscope.utils.logger import get_logger
logger = get_logger()
class Speech2Diarization:
    """Speech2Diarlization class
@@ -211,7 +208,6 @@
            if isinstance(raw_inputs, torch.Tensor):
                raw_inputs = raw_inputs.numpy()
            data_path_and_name_and_type = [raw_inputs[0], "speech", "bytes"]
        logger.info(data_path_and_name_and_type)
        loader = EENDOLADiarTask.build_streaming_iterator(
            data_path_and_name_and_type,
            dtype=dtype,
funasr/models/e2e_diar_eend_ola.py
@@ -190,8 +190,6 @@
                            shuffle: bool = True,
                            threshold: float = 0.5,
                            **kwargs):
        if self.frontend is not None:
            speech = self.frontend(speech)
        speech = [s[:s_len] for s, s_len in zip(speech, speech_lengths)]
        emb = self.forward_encoder(speech, speech_lengths)
        if shuffle: