From 74a00eecd7e0f90766f9fcd709f3e287ad6d97b3 Mon Sep 17 00:00:00 2001 From: Max Bain Date: Mon, 5 Jun 2023 15:33:04 +0100 Subject: [PATCH] suppress numerals fix --- whisperx/asr.py | 12 +----------- 1 file changed, 1 insertion(+), 11 deletions(-) diff --git a/whisperx/asr.py b/whisperx/asr.py index f2a7203..501b21d 100644 --- a/whisperx/asr.py +++ b/whisperx/asr.py @@ -11,7 +11,7 @@ from transformers.pipelines.pt_utils import PipelineIterator from .audio import N_SAMPLES, SAMPLE_RATE, load_audio, log_mel_spectrogram from .vad import load_vad_model, merge_chunks -<<<<<<< HEAD +from .types import TranscriptionResult, SingleSegment def find_numeral_symbol_tokens(tokenizer): numeral_symbol_tokens = [] @@ -21,12 +21,6 @@ def find_numeral_symbol_tokens(tokenizer): numeral_symbol_tokens.append(i) return numeral_symbol_tokens - -def load_model(whisper_arch, device, compute_type="float16", asr_options=None, language=None, - vad_options=None, model=None, task="transcribe"): -======= -from .types import TranscriptionResult, SingleSegment - def load_model(whisper_arch, device, device_index=0, @@ -37,7 +31,6 @@ def load_model(whisper_arch, model=None, task="transcribe", download_root=None): ->>>>>>> ec6a110cdf2616919cfd0a616f9ae2fbdd44903f '''Load a Whisper model for inference. Args: whisper_arch: str - The name of the Whisper model to load. @@ -100,9 +93,6 @@ def load_model(whisper_arch, default_asr_options = faster_whisper.transcribe.TranscriptionOptions(**default_asr_options) - - - default_vad_options = { "vad_onset": 0.500, "vad_offset": 0.363