LAP-DEV commited on
Commit
5999e47
·
verified ·
1 Parent(s): 7d69fde

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=self.format_suppress_tokens_str(params.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,