chrissewell/whisperX
1
0
Fork 0
mirror of https://github.com/m-bain/whisperX.git synced 2025-07-01 18:17:27 -04:00
Code Issues Packages Projects Releases Wiki Activity

fix: ZeroDivisionError when --print_progress True

Browse Source
This commit is contained in:
Manohar Reddy
2023-09-27 20:10:43 +05:30
parent 2a11ce3ef0
commit a0b6459c8b
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
whisperx/alignment.py
View File

@ -100,7 +100,6 @@ def align(
return_char_alignments: bool = False, return_char_alignments: bool = False,
print_progress: bool = False, print_progress: bool = False,
combined_progress: bool = False, combined_progress: bool = False,
total_segments: int = 0
) -> AlignedTranscriptionResult: ) -> AlignedTranscriptionResult:
""" """
Align phoneme recognition predictions to known transcription. Align phoneme recognition predictions to known transcription.
@ -120,6 +119,7 @@ def align(
model_type = align_model_metadata["type"] model_type = align_model_metadata["type"]
# 1. Preprocess to keep only characters in dictionary # 1. Preprocess to keep only characters in dictionary
total_segments = len(transcript)
for sdx, segment in enumerate(transcript): for sdx, segment in enumerate(transcript):
# strip spaces at beginning / end, but keep track of the amount. # strip spaces at beginning / end, but keep track of the amount.
if print_progress: if print_progress: