From 520796a34c46f7e6c56a65b50d9c6e09e2637b5d Mon Sep 17 00:00:00 2001 From: VulumeCode <2590984+VulumeCode@users.noreply.github.com> Date: Mon, 26 Sep 2022 13:35:21 +0200 Subject: [PATCH] fix token suppression (#123) --- whisper/decoding.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/whisper/decoding.py b/whisper/decoding.py index bd6322848..891aaa47b 100644 --- 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]