fix token suppression (openai#123)

MiscellaneousStuff · Sep 26, 2022 · 520796a · 520796a
1 parent ead77fa
commit 520796a
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/whisper/decoding.py b/whisper/decoding.py
@@ -537,8 +537,7 @@ def _get_suppress_tokens(self) -> Tuple[int]:
  elif suppress_tokens is None or len(suppress_tokens) == 0:
  suppress_tokens = [] # interpret empty string as an empty list
  else:
- assert isinstance(self.options.suppress_tokens, list), "suppress_tokens must be a list"
- suppress_tokens = self.options.suppress_tokens
+ assert isinstance(suppress_tokens, list), "suppress_tokens must be a list"
 
  suppress_tokens.extend(
  [self.tokenizer.sot, self.tokenizer.sot_prev, self.tokenizer.sot_lm]