mirrors/whispering
1
0
Fork 0
mirror of https://github.com/shirayu/whispering.git synced 2025-03-31 14:29:35 +00:00
Code Issues Projects Releases Packages Wiki Activity

Add debug log

Browse source
This commit is contained in:
Yuta Hayashibe 2022-09-24 15:27:02 +09:00
parent 146fc02f44
commit f75b13e0ba
1 changed files with 9 additions and 5 deletions

14
whisper_streaming/transcriber.py
View file

@ -241,15 +241,16 @@ class WhisperStreamingTranscriber:
seek: int = 0 seek: int = 0
rest_start: Optional[int] = None rest_start: Optional[int] = None
while seek < mel.shape[-1]: while seek < mel.shape[-1]:
logger.debug(
f"seek={seek}, timestamp={self.timestamp}, rest_start={rest_start}"
)
segment = ( segment = (
pad_or_trim(mel[:, :, seek:], N_FRAMES) pad_or_trim(mel[:, :, seek:], N_FRAMES)
.to(self.model.device) # type: ignore .to(self.model.device) # type: ignore
.to(self.dtype) .to(self.dtype)
) )
logger.debug(
f"seek={seek}, timestamp={self.timestamp}, rest_start={rest_start},"
f"mel.shape: {mel.shape}, segment.shape: {segment.shape}"
)
results = self._decode_with_fallback( results = self._decode_with_fallback(
segment=segment, segment=segment,
) )
@ -266,7 +267,9 @@ class WhisperStreamingTranscriber:
): ):
seek += segment.shape[-1] seek += segment.shape[-1]
rest_start = None rest_start = None
logger.debug(f"Skip: {segment.shape[-1]}") logger.debug(
f"Skip: {segment.shape[-1]}, new seek={seek}, mel.shape: {mel.shape}"
)
continue continue
segment_duration = segment.shape[-1] * HOP_LENGTH / SAMPLE_RATE segment_duration = segment.shape[-1] * HOP_LENGTH / SAMPLE_RATE
@ -284,8 +287,9 @@ class WhisperStreamingTranscriber:
else: else:
seek += last_timestamp_position * self.input_stride seek += last_timestamp_position * self.input_stride
rest_start = seek rest_start = seek
logger.debug(f"new seek={seek}, mel.shape: {mel.shape}")
logger.debug(f"Last rest_start={rest_start}") logger.debug(f"Last rest_start={rest_start}, mel.shape: {mel.shape}")
if rest_start is None: if rest_start is None:
return return