Fix bug when searching for vocoder

pull/470/head
babysor00 3 years ago
parent fc26c38152
commit 3c97d22938

@ -453,6 +453,7 @@ class Toolbox:
model_config_fpaths = list(model_fpath.parent.rglob("*.json")) model_config_fpaths = list(model_fpath.parent.rglob("*.json"))
if self.vc_mode and self.ui.current_extractor_fpath is None: if self.vc_mode and self.ui.current_extractor_fpath is None:
return return
if len(model_config_fpaths) > 0:
model_config_fpath = model_config_fpaths[0] model_config_fpath = model_config_fpaths[0]
else: else:
vocoder = rnn_vocoder vocoder = rnn_vocoder

@ -19,7 +19,7 @@ def load_checkpoint(filepath, device):
return checkpoint_dict return checkpoint_dict
def load_model(weights_fpath, config_fpath="./vocoder/saved_models/24k/config.json", verbose=True): def load_model(weights_fpath, config_fpath="./vocoder/hifigan/config_16k.json", verbose=True):
global generator, _device, output_sample_rate global generator, _device, output_sample_rate
if verbose: if verbose:

Loading…
Cancel
Save