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

Remove unused import

Browse Source
This commit is contained in:
Guillaume Klein
2023-03-18 09:47:02 +01:00
parent d82be59d5f
commit a70aac18ae
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
faster_whisper/transcribe.py
View File

@@ -1,4 +1,3 @@
import collections
import itertools import itertools
import os import os
import zlib import zlib
@@ -445,6 +444,7 @@ class WhisperModel:
for temperature in options.temperatures: for temperature in options.temperatures:
if temperature > 0: if temperature > 0:
print(temperature)
kwargs = { kwargs = {
"beam_size": 1, "beam_size": 1,
"num_hypotheses": options.best_of, "num_hypotheses": options.best_of,
@@ -472,6 +472,7 @@ class WhisperModel:
)[0] )[0]
tokens = result.sequences_ids[0] tokens = result.sequences_ids[0]
print(tokens)
# Recover the average log prob from the returned score. # Recover the average log prob from the returned score.
seq_len = len(tokens) seq_len = len(tokens)