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

Add default path for hifi

Browse Source
This commit is contained in:
babysor00 2022-03-08 09:17:56 +08:00
parent 3c97d22938
commit 79f424d614
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
vocoder/hifigan/inference.py
View File

@ -19,12 +19,14 @@ def load_checkpoint(filepath, device):
return checkpoint_dict return checkpoint_dict
def load_model(weights_fpath, config_fpath="./vocoder/hifigan/config_16k.json", verbose=True): def load_model(weights_fpath, config_fpath, verbose=True):
global generator, _device, output_sample_rate global generator, _device, output_sample_rate
if verbose: if verbose:
print("Building hifigan") print("Building hifigan")
if config_fpath == None:
config_fpath = "./vocoder/hifigan/config_16k_.json"
with open(config_fpath) as f: with open(config_fpath) as f:
data = f.read() data = f.read()
json_config = json.loads(data) json_config = json.loads(data)