Update modules/whisper/faster_whisper_inference.py
Browse files
modules/whisper/faster_whisper_inference.py
CHANGED
@@ -63,6 +63,7 @@ class FasterWhisperInference(WhisperBase):
|
|
63 |
start_time = time.time()
|
64 |
|
65 |
params = WhisperParameters.as_value(*whisper_params)
|
|
|
66 |
|
67 |
if params.model_size != self.current_model_size or self.model is None or self.current_compute_type != params.compute_type:
|
68 |
self.update_model(params.model_size, params.compute_type, progress)
|
@@ -84,7 +85,7 @@ class FasterWhisperInference(WhisperBase):
|
|
84 |
no_repeat_ngram_size=params.no_repeat_ngram_size,
|
85 |
prefix=params.prefix,
|
86 |
suppress_blank=params.suppress_blank,
|
87 |
-
suppress_tokens=
|
88 |
max_initial_timestamp=params.max_initial_timestamp,
|
89 |
word_timestamps=params.word_timestamps,
|
90 |
prepend_punctuations=params.prepend_punctuations,
|
|
|
63 |
start_time = time.time()
|
64 |
|
65 |
params = WhisperParameters.as_value(*whisper_params)
|
66 |
+
params.suppress_tokens = self.format_suppress_tokens_str(params.suppress_tokens)
|
67 |
|
68 |
if params.model_size != self.current_model_size or self.model is None or self.current_compute_type != params.compute_type:
|
69 |
self.update_model(params.model_size, params.compute_type, progress)
|
|
|
85 |
no_repeat_ngram_size=params.no_repeat_ngram_size,
|
86 |
prefix=params.prefix,
|
87 |
suppress_blank=params.suppress_blank,
|
88 |
+
suppress_tokens=params.suppress_tokens,
|
89 |
max_initial_timestamp=params.max_initial_timestamp,
|
90 |
word_timestamps=params.word_timestamps,
|
91 |
prepend_punctuations=params.prepend_punctuations,
|