From 11fd8ab3011dedd05d83e215a90cda63da7de653 Mon Sep 17 00:00:00 2001 From: Purfview <69023953+Purfview@users.noreply.github.com> Date: Sun, 29 Dec 2024 11:38:58 +0000 Subject: [PATCH] Fix neg_threshold (#1191) --- faster_whisper/vad.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/faster_whisper/vad.py b/faster_whisper/vad.py index 1f7d2057..87e898c1 100644 --- a/faster_whisper/vad.py +++ b/faster_whisper/vad.py @@ -35,7 +35,7 @@ class VadOptions: """ threshold: float = 0.5 - neg_threshold: float = threshold - 0.15 + neg_threshold: float = None min_speech_duration_ms: int = 0 max_speech_duration_s: float = float("inf") min_silence_duration_ms: int = 2000 @@ -63,6 +63,7 @@ def get_speech_timestamps( vad_options = VadOptions(**kwargs) threshold = vad_options.threshold + neg_threshold = vad_options.neg_threshold min_speech_duration_ms = vad_options.min_speech_duration_ms max_speech_duration_s = vad_options.max_speech_duration_s min_silence_duration_ms = vad_options.min_silence_duration_ms @@ -90,7 +91,8 @@ def get_speech_timestamps( triggered = False speeches = [] current_speech = {} - neg_threshold = vad_options.neg_threshold + if neg_threshold is None: + neg_threshold = max(threshold - 0.15, 0.01) # to save potential segment end (and tolerate some silence) temp_end = 0