speech_recognition/FunASR
1
0
Fork 0
mirror of https://github.com/modelscope/FunASR synced 2025-09-15 14:48:36 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Update asr_infer.py

Browse Source 
rename simu_streaming to fake_streaming
This commit is contained in:
aky15 2023-09-14 16:08:53 +08:00 committed by GitHub
parent d13baaea9c
commit 11d5964951
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
funasr/bin/asr_infer.py
View File

@ -1337,7 +1337,7 @@ class Speech2TextTransducer:
quantize_dtype: str = "qint8", quantize_dtype: str = "qint8",
nbest: int = 1, nbest: int = 1,
streaming: bool = False, streaming: bool = False,
simu_streaming: bool = False, fake_streaming: bool = False,
full_utt: bool = False, full_utt: bool = False,
chunk_size: int = 16, chunk_size: int = 16,
left_context: int = 32, left_context: int = 32,
@ -1432,7 +1432,7 @@ class Speech2TextTransducer:
self.beam_search = beam_search self.beam_search = beam_search
self.streaming = streaming self.streaming = streaming
self.simu_streaming = simu_streaming self.fake_streaming = fake_streaming
self.full_utt = full_utt self.full_utt = full_utt
self.chunk_size = max(chunk_size, 0) self.chunk_size = max(chunk_size, 0)
self.left_context = left_context self.left_context = left_context
@ -1442,8 +1442,8 @@ class Speech2TextTransducer:
self.streaming = False self.streaming = False
self.asr_model.encoder.dynamic_chunk_training = False self.asr_model.encoder.dynamic_chunk_training = False
if not simu_streaming or chunk_size == 0: if not fake_streaming or chunk_size == 0:
self.simu_streaming = False self.fake_streaming = False
self.asr_model.encoder.dynamic_chunk_training = False self.asr_model.encoder.dynamic_chunk_training = False
self.frontend = frontend self.frontend = frontend
@ -1520,7 +1520,7 @@ class Speech2TextTransducer:
return nbest_hyps return nbest_hyps
@torch.no_grad() @torch.no_grad()
def simu_streaming_decode(self, speech: Union[torch.Tensor, np.ndarray]) -> List[HypothesisTransducer]: def fake_streaming_decode(self, speech: Union[torch.Tensor, np.ndarray]) -> List[HypothesisTransducer]:
"""Speech2Text call. """Speech2Text call.
Args: Args:
speech: Speech data. (S) speech: Speech data. (S)