zhangyiss/mlx-examples
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx-examples.git synced 2025-06-24 01:17:28 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix looping in whisper (#1273)

Browse Source
This commit is contained in:
Awni Hannun 2025-02-10 13:04:35 -08:00 committed by GitHub
parent 5865899c81
commit bded1a8fcd
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
whisper/mlx_whisper/transcribe.py
View File

@ -195,6 +195,8 @@ def transcribe(
seek_points.append(0) seek_points.append(0)
if len(seek_points) % 2 == 1: if len(seek_points) % 2 == 1:
seek_points.append(content_frames) seek_points.append(content_frames)
else:
seek_points[-1] = min(content_frames, seek_points[-1])
seek_clips: List[Tuple[int, int]] = list(zip(seek_points[::2], seek_points[1::2])) seek_clips: List[Tuple[int, int]] = list(zip(seek_points[::2], seek_points[1::2]))
punctuation = "\"'“¿([{-\"'.。,，!！?？:：”)]}、" punctuation = "\"'“¿([{-\"'.。,，!！?？:：”)]}、"