Spaces:
Running
Running
File size: 3,290 Bytes
5a30378 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 | # Copyright (c) 2023 Amphion.
#
# This source code is licensed under the MIT license found in the
# LICENSE file in the root directory of this source tree.
import os
import json5
# removed all training/logging utilities (str2bool, pad_f0_to_tensors, ValueWindow, Logger,
# get_filelist, etc.) bcs they weren't needed for inference. kept only load_config and JsonHParams.
def override_config(base_config, new_config):
"""Update new configurations in the original dict with the new dict
Args:
base_config (dict): original dict to be overridden
new_config (dict): dict with new configurations
Returns:
dict: updated configuration dict
"""
for k, v in new_config.items():
if type(v) == dict:
if k not in base_config.keys():
base_config[k] = {}
base_config[k] = override_config(base_config[k], v)
else:
base_config[k] = v
return base_config
def get_lowercase_keys_config(cfg):
"""Change all keys in cfg to lower case
Args:
cfg (dict): dictionary that stores configurations
Returns:
dict: dictionary that stores configurations
"""
updated_cfg = dict()
for k, v in cfg.items():
if type(v) == dict:
v = get_lowercase_keys_config(v)
updated_cfg[k.lower()] = v
return updated_cfg
def _load_config(config_fn, lowercase=False):
"""Load configurations into a dictionary
Args:
config_fn (str): path to configuration file
lowercase (bool, optional): whether changing keys to lower case. Defaults to False.
Returns:
dict: dictionary that stores configurations
"""
with open(config_fn, "r") as f:
data = f.read()
config_ = json5.loads(data)
if "base_config" in config_:
try:
p_config_path = os.path.join(os.getenv("WORK_DIR"), config_["base_config"])
except:
p_config_path = config_["base_config"]
p_config_ = _load_config(p_config_path)
config_ = override_config(p_config_, config_)
if lowercase:
config_ = get_lowercase_keys_config(config_)
return config_
def load_config(config_fn, lowercase=False):
"""Load configurations into a dictionary
Args:
config_fn (str): path to configuration file
lowercase (bool, optional): _description_. Defaults to False.
Returns:
JsonHParams: an object that stores configurations
"""
config_ = _load_config(config_fn, lowercase=lowercase)
cfg = JsonHParams(**config_)
return cfg
class JsonHParams:
def __init__(self, **kwargs):
for k, v in kwargs.items():
if type(v) == dict:
v = JsonHParams(**v)
self[k] = v
def keys(self):
return self.__dict__.keys()
def items(self):
return self.__dict__.items()
def values(self):
return self.__dict__.values()
def __len__(self):
return len(self.__dict__)
def __getitem__(self, key):
return getattr(self, key)
def __setitem__(self, key, value):
return setattr(self, key, value)
def __contains__(self, key):
return key in self.__dict__
def __repr__(self):
return self.__dict__.__repr__()
|