mirrors/whispering
1
0
Fork 0
mirror of https://github.com/shirayu/whispering.git synced 2024-06-12 18:29:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

Improved logger

Browse source
This commit is contained in:
Yuta Hayashibe 2022-09-23 22:11:36 +09:00
parent 111e3a94a6
commit 3725694cbb
2 changed files with 7 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
whisper_streaming/cli.py
View file

@ -3,7 +3,7 @@
import argparse import argparse
import queue import queue
import sys import sys
from logging import DEBUG, INFO, StreamHandler, getLogger from logging import DEBUG, INFO, StreamHandler, basicConfig, getLogger
from typing import Optional, Union from typing import Optional, Union
import sounddevice as sd import sounddevice as sd
@ -94,13 +94,11 @@ def get_opts() -> argparse.Namespace:
def main() -> None: def main() -> None:
opts = get_opts() opts = get_opts()
basicConfig(
level=DEBUG if opts.debug else INFO,
format="[%(asctime)s] %(module)s.%(funcName)s %(levelname)s -> %(message)s",
)
root_logger = getLogger()
root_logger.addHandler(StreamHandler(sys.stderr))
if opts.debug:
root_logger.setLevel(DEBUG)
else:
root_logger.setLevel(INFO)
if opts.beam_size <= 0: if opts.beam_size <= 0:
opts.beam_size = None opts.beam_size = None
try: try:

4
whisper_streaming/transcriber.py
View file

@ -195,7 +195,7 @@ class WhisperStreamingTranscriber:
) )
if chunk is not None: if chunk is not None:
yield chunk yield chunk
self.timestamp += float(segment_duration * HOP_LENGTH / SAMPLE_RATE) self.timestamp += duration
if result.temperature > 0.5: if result.temperature > 0.5:
# do not feed the prompt tokens if a high temperature was used # do not feed the prompt tokens if a high temperature was used
@ -217,7 +217,7 @@ 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(seek) logger.debug(f"seek={seek}, timestamp={self.timestamp}")
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