Rename model

modules/whisper/whisper_base.py

@@ -514,7 +514,7 @@ class WhisperBase(ABC):
 
     @staticmethod
     def cache_parameters(
-        whisper_params: WhisperValues,
+        whisper_params: TranscriptionPipelineParams,
         add_timestamp: bool
     ):
         """cache parameters to the yaml file"""

tests/test_bgm_separation.py

@@ -1,6 +1,6 @@
 from modules.utils.paths import *
 from modules.whisper.whisper_factory import WhisperFactory
-from modules.whisper.data_classes import WhisperValues
+from modules.whisper.data_classes import TranscriptionPipelineParams
 from test_config import *
 from test_transcription import download_file, test_transcribe
 

tests/test_diarization.py

@@ -1,6 +1,6 @@
 from modules.utils.paths import *
 from modules.whisper.whisper_factory import WhisperFactory
-from modules.whisper.data_classes import WhisperValues
+from modules.whisper.data_classes import TranscriptionPipelineParams
 from test_config import *
 from test_transcription import download_file, test_transcribe
 

tests/test_transcription.py

@@ -1,5 +1,5 @@
 from modules.whisper.whisper_factory import WhisperFactory
-from modules.whisper.data_classes import WhisperValues
+from modules.whisper.data_classes import TranscriptionPipelineParams
 from modules.utils.paths import WEBUI_DIR
 from test_config import *
 
@@ -37,7 +37,7 @@ def test_transcribe(
         f"""Diarization Device: {whisper_inferencer.diarizer.device}"""
     )
 
-    hparams = WhisperValues(
+    hparams = TranscriptionPipelineParams(
         model_size=TEST_WHISPER_MODEL,
         vad_filter=vad_filter,
         is_bgm_separate=bgm_separation,

tests/test_vad.py

@@ -1,6 +1,6 @@
 from modules.utils.paths import *
 from modules.whisper.whisper_factory import WhisperFactory
-from modules.whisper.data_classes import WhisperValues
+from modules.whisper.data_classes import TranscriptionPipelineParams
 from test_config import *
 from test_transcription import download_file, test_transcribe