add version control

pull/3927/head
zxcd 10 months ago
parent 6cbb8878b4
commit edb852e33a

@ -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)

Loading…
Cancel
Save