heimoshuiyu/faster-whisper
1
0
Fork 0
Code Issues Pull Requests 1 Packages Projects Releases Wiki Activity

Minor code simplification

Browse Source
This commit is contained in:
Guillaume Klein
2023-02-22 11:02:11 +01:00
parent 4b8237da1b
commit d91365e321
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
faster_whisper/transcribe.py
View File

@@ -332,12 +332,13 @@ class WhisperModel:
**kwargs, **kwargs,
)[0] )[0]
tokens = result.sequences_ids[0]
# Recover the average log prob from the returned score. # Recover the average log prob from the returned score.
seq_len = len(result.sequences_ids[0]) seq_len = len(tokens)
cum_log_prob = result.scores[0] * (seq_len**options.length_penalty) cum_log_prob = result.scores[0] * (seq_len**options.length_penalty)
avg_log_prob = cum_log_prob / (seq_len + 1) avg_log_prob = cum_log_prob / (seq_len + 1)
tokens = result.sequences_ids[0]
text = self.decode_text_tokens(tokens).strip() text = self.decode_text_tokens(tokens).strip()
compression_ratio = get_compression_ratio(text) compression_ratio = get_compression_ratio(text)