Mirroristas/whisper
2
0
Fork 0
mirror of https://github.com/openai/whisper.git synced 2025-11-29 16:38:40 +00:00
Code Issues Packages Projects Releases Wiki Activity

Do not assert for a negative timestamp

Browse Source 
This may be thrown when `verbose` is set to True and a very short audio is transcribed, resulting in a negative timestamp. Even though this is not a desired use case, it is preferable to keep going without raising an exception.
This commit is contained in:
Gaspard Petit 2024-01-07 21:33:21 -05:00
parent ba3f3cd54b
commit 17de1f20c0
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
whisper/utils.py
View File

@ -50,7 +50,9 @@ def compression_ratio(text) -> float:
def format_timestamp( def format_timestamp(
seconds: float, always_include_hours: bool = False, decimal_marker: str = "." seconds: float, always_include_hours: bool = False, decimal_marker: str = "."
): ):
assert seconds >= 0, "non-negative timestamp expected" sign = "-" if seconds < 0 else ""
seconds = abs(seconds)
milliseconds = round(seconds * 1000.0) milliseconds = round(seconds * 1000.0)
hours = milliseconds // 3_600_000 hours = milliseconds // 3_600_000
@ -64,7 +66,7 @@ def format_timestamp(
hours_marker = f"{hours:02d}:" if always_include_hours or hours > 0 else "" hours_marker = f"{hours:02d}:" if always_include_hours or hours > 0 else ""
return ( return (
f"{hours_marker}{minutes:02d}:{seconds:02d}{decimal_marker}{milliseconds:03d}" f"{sign}{hours_marker}{minutes:02d}:{seconds:02d}{decimal_marker}{milliseconds:03d}"
) )