Spaces:
Running
Running
jhj0517
commited on
Commit
·
aa98cf2
1
Parent(s):
6db9d8d
Fix param validation
Browse files
modules/whisper/whisper_base.py
CHANGED
|
@@ -15,7 +15,7 @@ from dataclasses import astuple
|
|
| 15 |
from modules.uvr.music_separator import MusicSeparator
|
| 16 |
from modules.utils.paths import (WHISPER_MODELS_DIR, DIARIZATION_MODELS_DIR, OUTPUT_DIR, DEFAULT_PARAMETERS_CONFIG_PATH,
|
| 17 |
UVR_MODELS_DIR)
|
| 18 |
-
from modules.utils.constants import AUTOMATIC_DETECTION
|
| 19 |
from modules.utils.subtitle_manager import get_srt, get_vtt, get_txt, write_file, safe_filename
|
| 20 |
from modules.utils.youtube_manager import get_ytdata, get_ytaudio
|
| 21 |
from modules.utils.files_manager import get_media_files, format_gradio_files, load_yaml, save_yaml
|
|
@@ -127,10 +127,6 @@ class WhisperBase(ABC):
|
|
| 127 |
self.music_separator.offload()
|
| 128 |
|
| 129 |
if vad_params.vad_filter:
|
| 130 |
-
# Explicit value set for float('inf') from gr.Number()
|
| 131 |
-
if vad_params.max_speech_duration_s is None or vad_params.max_speech_duration_s >= 9999:
|
| 132 |
-
vad_params.max_speech_duration_s = float('inf')
|
| 133 |
-
|
| 134 |
vad_options = VadOptions(
|
| 135 |
threshold=vad_params.threshold,
|
| 136 |
min_speech_duration_ms=vad_params.min_speech_duration_ms,
|
|
@@ -535,8 +531,8 @@ class WhisperBase(ABC):
|
|
| 535 |
params.whisper.hallucination_silence_threshold = None
|
| 536 |
if params.whisper.language_detection_threshold == 0:
|
| 537 |
params.whisper.language_detection_threshold = None
|
| 538 |
-
if params.
|
| 539 |
-
params.
|
| 540 |
return params
|
| 541 |
|
| 542 |
@staticmethod
|
|
|
|
| 15 |
from modules.uvr.music_separator import MusicSeparator
|
| 16 |
from modules.utils.paths import (WHISPER_MODELS_DIR, DIARIZATION_MODELS_DIR, OUTPUT_DIR, DEFAULT_PARAMETERS_CONFIG_PATH,
|
| 17 |
UVR_MODELS_DIR)
|
| 18 |
+
from modules.utils.constants import AUTOMATIC_DETECTION
|
| 19 |
from modules.utils.subtitle_manager import get_srt, get_vtt, get_txt, write_file, safe_filename
|
| 20 |
from modules.utils.youtube_manager import get_ytdata, get_ytaudio
|
| 21 |
from modules.utils.files_manager import get_media_files, format_gradio_files, load_yaml, save_yaml
|
|
|
|
| 127 |
self.music_separator.offload()
|
| 128 |
|
| 129 |
if vad_params.vad_filter:
|
|
|
|
|
|
|
|
|
|
|
|
|
| 130 |
vad_options = VadOptions(
|
| 131 |
threshold=vad_params.threshold,
|
| 132 |
min_speech_duration_ms=vad_params.min_speech_duration_ms,
|
|
|
|
| 531 |
params.whisper.hallucination_silence_threshold = None
|
| 532 |
if params.whisper.language_detection_threshold == 0:
|
| 533 |
params.whisper.language_detection_threshold = None
|
| 534 |
+
if params.vad.max_speech_duration_s >= 9999:
|
| 535 |
+
params.vad.max_speech_duration_s = float('inf')
|
| 536 |
return params
|
| 537 |
|
| 538 |
@staticmethod
|