Spaces:
Running
Running
jhj0517
commited on
Commit
·
813ca79
1
Parent(s):
289ab0d
Add yaml utils
Browse files
modules/whisper/whisper_base.py
CHANGED
@@ -12,7 +12,7 @@ from dataclasses import astuple
|
|
12 |
from modules.utils.paths import (WHISPER_MODELS_DIR, DIARIZATION_MODELS_DIR, OUTPUT_DIR, DEFAULT_PARAMETERS_CONFIG_PATH)
|
13 |
from modules.utils.subtitle_manager import get_srt, get_vtt, get_txt, write_file, safe_filename
|
14 |
from modules.utils.youtube_manager import get_ytdata, get_ytaudio
|
15 |
-
from modules.utils.files_manager import get_media_files, format_gradio_files
|
16 |
from modules.whisper.whisper_parameter import *
|
17 |
from modules.diarize.diarizer import Diarizer
|
18 |
from modules.vad.silero_vad import SileroVAD
|
@@ -440,7 +440,8 @@ class WhisperBase(ABC):
|
|
440 |
|
441 |
@staticmethod
|
442 |
def cache_parameters(whisper_params: WhisperValues):
|
443 |
-
|
|
|
|
|
444 |
|
445 |
-
|
446 |
-
file.write(cached_yaml)
|
|
|
12 |
from modules.utils.paths import (WHISPER_MODELS_DIR, DIARIZATION_MODELS_DIR, OUTPUT_DIR, DEFAULT_PARAMETERS_CONFIG_PATH)
|
13 |
from modules.utils.subtitle_manager import get_srt, get_vtt, get_txt, write_file, safe_filename
|
14 |
from modules.utils.youtube_manager import get_ytdata, get_ytaudio
|
15 |
+
from modules.utils.files_manager import get_media_files, format_gradio_files, load_yaml, save_yaml
|
16 |
from modules.whisper.whisper_parameter import *
|
17 |
from modules.diarize.diarizer import Diarizer
|
18 |
from modules.vad.silero_vad import SileroVAD
|
|
|
440 |
|
441 |
@staticmethod
|
442 |
def cache_parameters(whisper_params: WhisperValues):
|
443 |
+
cached_params = load_yaml(DEFAULT_PARAMETERS_CONFIG_PATH)
|
444 |
+
cached_whisper_param = whisper_params.to_yaml()
|
445 |
+
cached_yaml = {**cached_params, **cached_whisper_param}
|
446 |
|
447 |
+
save_yaml(cached_yaml, DEFAULT_PARAMETERS_CONFIG_PATH)
|
|