StarMirror/MockingBird
1
0
Fork 0
mirror of https://github.com/babysor/MockingBird.git synced 2024-03-22 13:11:31 +08:00
Code Issues Projects Releases Wiki Activity

Fix bug when searching for vocoder

Browse Source
This commit is contained in:
babysor00 2022-03-07 20:08:29 +08:00
parent fc26c38152
commit 3c97d22938
2 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
toolbox/__init__.py
View File

@ -453,7 +453,8 @@ 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
model_config_fpath = model_config_fpaths[0] if len(model_config_fpaths) > 0:
model_config_fpath = model_config_fpaths[0]
else: else:
vocoder = rnn_vocoder vocoder = rnn_vocoder
self.ui.log("set wavernn as vocoder") self.ui.log("set wavernn as vocoder")

2
vocoder/hifigan/inference.py
View File

@ -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: