Refactor: replace utils.HParams with HyperParameters Pydantic model

HyperParameters is largely a drop-in replacement for utils.HParams, which ensures type safety for hyper-parameters.
This commit is contained in:
tsukumi
2024-03-08 15:52:37 +00:00
parent 7f0b252806
commit a84783a6cc
11 changed files with 190 additions and 221 deletions

View File

@@ -8,6 +8,7 @@ from tqdm import tqdm
from style_bert_vits2.logging import logger
from style_bert_vits2.models import utils
from style_bert_vits2.models.hyper_parameters import HyperParameters
from style_bert_vits2.utils.stdout_wrapper import SAFE_STDOUT
from config import config
@@ -72,7 +73,7 @@ if __name__ == "__main__":
config_path = args.config
num_processes = args.num_processes
hps = utils.get_hparams_from_file(config_path)
hps = HyperParameters.load_from_json(config_path)
device = config.style_gen_config.device