chrissewell/whisperX
1
0
Fork 0
mirror of https://github.com/m-bain/whisperX.git synced 2025-07-01 18:17:27 -04:00
Code Issues Packages Projects Releases Wiki Activity

dataclasses replace method

Browse Source
This commit is contained in:
justintennenbaum
2025-01-07 12:35:41 -05:00
committed by Barabazs
parent 10b05fc43f
commit a90bd1ce3f
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
whisperx/asr.py
View File

@ -1,6 +1,7 @@
import os import os
import warnings import warnings
from typing import List, NamedTuple, Optional, Union from typing import List, NamedTuple, Optional, Union
from dataclasses import replace
import ctranslate2 import ctranslate2
import faster_whisper import faster_whisper
@ -240,7 +241,7 @@ class FasterWhisperPipeline(Pipeline):
print(f"Suppressing numeral and symbol tokens") print(f"Suppressing numeral and symbol tokens")
new_suppressed_tokens = numeral_symbol_tokens + self.options.suppress_tokens new_suppressed_tokens = numeral_symbol_tokens + self.options.suppress_tokens
new_suppressed_tokens = list(set(new_suppressed_tokens)) new_suppressed_tokens = list(set(new_suppressed_tokens))
self.options = self.options._replace(suppress_tokens=new_suppressed_tokens) self.options = replace(self.options, suppress_tokens=new_suppressed_tokens)
segments: List[SingleSegment] = [] segments: List[SingleSegment] = []
batch_size = batch_size or self._batch_size batch_size = batch_size or self._batch_size
@ -269,7 +270,7 @@ class FasterWhisperPipeline(Pipeline):
# revert suppressed tokens if suppress_numerals is enabled # revert suppressed tokens if suppress_numerals is enabled
if self.suppress_numerals: if self.suppress_numerals:
self.options = self.options._replace(suppress_tokens=previous_suppress_tokens) self.options = replace(self.options, suppress_tokens=previous_suppress_tokens)
return {"segments": segments, "language": language} return {"segments": segments, "language": language}