mirrors/whispering
1
0
Fork 0
mirror of https://github.com/shirayu/whispering.git synced 2024-11-14 04:31:34 +00:00
Code Issues Projects Releases Packages Wiki Activity

Add debug log

Browse source
This commit is contained in:
Yuta Hayashibe 2022-09-23 22:31:26 +09:00
parent b51d7c6cce
commit 56508bc282
1 changed files with 3 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
whisper_streaming/transcriber.py
View file

@ -150,8 +150,10 @@ class WhisperStreamingTranscriber:
if (
len(consecutive) > 0
): # if the output contains two consecutive timestamp tokens
logger.debug(f"Length of consecutive: {len(consecutive)}")
last_slice = 0
for current_slice in consecutive:
logger.debug(f" last_slice={last_slice}, current_slice={current_slice}")
sliced_tokens = tokens[last_slice:current_slice]
start_timestamp_position = (
sliced_tokens[0].item() - self.tokenizer.timestamp_begin
@ -177,6 +179,7 @@ class WhisperStreamingTranscriber:
self.timestamp += last_timestamp_position0 * self.time_precision
yield last_timestamp_position0
else:
logger.debug("Length of consecutive: 0")
duration = segment_duration
timestamps = tokens[timestamp_tokens.nonzero().flatten()]
if len(timestamps) > 0: