Make vad-related parameters configurable for batched inference. (SYST…

…RAN#923)
mobiusml · Oct 2, 2024 · 6b90b77 · 6b90b77
1 parent 5b2cc66
commit 6b90b77
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/faster_whisper/transcribe.py b/faster_whisper/transcribe.py
@@ -122,6 +122,8 @@ def __init__(
         device: Union[int, str, "torch.device"] = -1,
         chunk_length: int = 30,
         vad_device: Union[int, str, "torch.device"] = "auto",
+        vad_onset: float = 0.500,
+        vad_offset: float = 0.363,
         framework="pt",
         language: Optional[str] = None,
         **kwargs,
@@ -133,8 +135,8 @@ def __init__(
         self._batch_size = kwargs.pop("batch_size", None)
         self._num_workers = 0
         self.use_vad_model = use_vad_model
-        self.vad_onset = 0.500
-        self.vad_offset = 0.363
+        self.vad_onset = vad_onset
+        self.vad_offset = vad_offset
         self.vad_model_path = os.path.join(get_assets_path(), "pyannote_vad_model.bin")
         self.vad_model = None