LAP-DEV commited on
Commit
05447df
·
verified ·
1 Parent(s): 31b9662

Update modules/whisper/faster_whisper_inference.py

Browse files
modules/whisper/faster_whisper_inference.py CHANGED
@@ -187,12 +187,12 @@ class FasterWhisperInference(WhisperBase):
187
  else:
188
  return "auto"
189
 
190
- # @staticmethod
191
- # def format_suppress_tokens_str(suppress_tokens_str: str) -> List[int]:
192
- # try:
193
- # suppress_tokens = ast.literal_eval(suppress_tokens_str)
194
- # if not isinstance(suppress_tokens, list) or not all(isinstance(item, int) for item in suppress_tokens):
195
- # raise ValueError("Invalid Suppress Tokens. The value must be type of List[int]")
196
- # return suppress_tokens
197
- # except Exception as e:
198
- # raise ValueError("Invalid Suppress Tokens. The value must be type of List[int]")
 
187
  else:
188
  return "auto"
189
 
190
+ @staticmethod
191
+ def format_suppress_tokens_str(suppress_tokens_str: str) -> List[int]:
192
+ try:
193
+ suppress_tokens = ast.literal_eval(suppress_tokens_str)
194
+ if not isinstance(suppress_tokens, list) or not all(isinstance(item, int) for item in suppress_tokens):
195
+ raise ValueError("Invalid Suppress Tokens. The value must be type of List[int]")
196
+ return suppress_tokens
197
+ except Exception as e:
198
+ raise ValueError("Invalid Suppress Tokens. The value must be type of List[int]")