Remove unused import

This commit is contained in:
Guillaume Klein
2023-03-18 09:47:02 +01:00
parent d82be59d5f
commit a70aac18ae

View File

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