# we need to merge the text_config into the root level of hparams
def __init__(self, *args, **kwargs):
- hparams = Model.load_hparams(kwargs["dir_model"])
+ hparams = kwargs["hparams"] if "hparams" in kwargs else Model.load_hparams(args[0])
if "text_config" in hparams:
hparams = {**hparams, **hparams["text_config"]}
kwargs["hparams"] = hparams
# we need to merge the text_config into the root level of hparams
def __init__(self, *args, **kwargs):
- hparams = Model.load_hparams(kwargs["dir_model"])
+ hparams = kwargs["hparams"] if "hparams" in kwargs else Model.load_hparams(args[0])
if "text_config" in hparams:
hparams = {**hparams, **hparams["text_config"]}
kwargs["hparams"] = hparams
logger.error(f"Model {model_architecture} is not supported")
sys.exit(1)
- model_instance = model_class(dir_model=dir_model, ftype=output_type, fname_out=fname_out,
+ model_instance = model_class(dir_model, output_type, fname_out,
is_big_endian=args.bigendian, use_temp_file=args.use_temp_file,
eager=args.no_lazy,
metadata_override=args.metadata, model_name=args.model_name,