jhj0517 commited on
Commit
813ca79
·
1 Parent(s): 289ab0d

Add yaml utils

Browse files
Files changed (1) hide show
  1. modules/whisper/whisper_base.py +5 -4
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
- cached_yaml = whisper_params.to_yaml()
 
 
444
 
445
- with open(DEFAULT_PARAMETERS_CONFIG_PATH, 'w', encoding='utf-8') as file:
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)