|
|
@ -48,13 +48,27 @@ def main():
|
|
|
|
phone_vocab_path=args.phones_dict, tone_vocab_path=args.tones_dict)
|
|
|
|
phone_vocab_path=args.phones_dict, tone_vocab_path=args.tones_dict)
|
|
|
|
print("frontend done!")
|
|
|
|
print("frontend done!")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
# after paddle 3.0, support new inference interface
|
|
|
|
|
|
|
|
if paddle.__version__ >= '3.0.0' or paddle.__version__ == '0.0.0':
|
|
|
|
speedyspeech_config = inference.Config(
|
|
|
|
speedyspeech_config = inference.Config(
|
|
|
|
str(Path(args.inference_dir)), "speedyspeech")
|
|
|
|
str(Path(args.inference_dir)), "speedyspeech")
|
|
|
|
|
|
|
|
else:
|
|
|
|
|
|
|
|
speedyspeech_config = inference.Config(
|
|
|
|
|
|
|
|
str(Path(args.inference_dir) / "speedyspeech.pdmodel"),
|
|
|
|
|
|
|
|
str(Path(args.inference_dir) / "speedyspeech.pdiparams"))
|
|
|
|
|
|
|
|
|
|
|
|
speedyspeech_config.enable_use_gpu(100, 0)
|
|
|
|
speedyspeech_config.enable_use_gpu(100, 0)
|
|
|
|
speedyspeech_config.enable_memory_optim()
|
|
|
|
speedyspeech_config.enable_memory_optim()
|
|
|
|
speedyspeech_predictor = inference.create_predictor(speedyspeech_config)
|
|
|
|
speedyspeech_predictor = inference.create_predictor(speedyspeech_config)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
# after paddle 3.0, support new inference interface
|
|
|
|
|
|
|
|
if paddle.__version__ >= '3.0.0' or paddle.__version__ == '0.0.0':
|
|
|
|
pwg_config = inference.Config(str(Path(args.inference_dir)), "pwg")
|
|
|
|
pwg_config = inference.Config(str(Path(args.inference_dir)), "pwg")
|
|
|
|
|
|
|
|
else:
|
|
|
|
|
|
|
|
pwg_config = inference.Config(
|
|
|
|
|
|
|
|
str(Path(args.inference_dir) / "pwg.pdmodel"),
|
|
|
|
|
|
|
|
str(Path(args.inference_dir) / "pwg.pdiparams"))
|
|
|
|
|
|
|
|
|
|
|
|
pwg_config.enable_use_gpu(100, 0)
|
|
|
|
pwg_config.enable_use_gpu(100, 0)
|
|
|
|
pwg_config.enable_memory_optim()
|
|
|
|
pwg_config.enable_memory_optim()
|
|
|
|
pwg_predictor = inference.create_predictor(pwg_config)
|
|
|
|
pwg_predictor = inference.create_predictor(pwg_config)
|
|
|
|