msb-public
/
PaddleSpeech
mirror of https://github.com/PaddlePaddle/PaddleSpeech
1
0
Code Issues Projects Releases Wiki Activity

pir infer

Browse Source
pull/3926/head
zxcd 10 months ago
parent 67ae7c8dd2
commit 6cbb8878b4
1 changed files with 2 additions and 5 deletions
Show Stats Download Patch File Download Diff File

7
paddlespeech/t2s/exps/speedyspeech/inference.py
Unescape View File

@ -49,15 +49,12 @@ def main():
print("frontend done!") print("frontend done!")
speedyspeech_config = inference.Config( speedyspeech_config = inference.Config(
str(Path(args.inference_dir) / "speedyspeech.pdmodel"), str(Path(args.inference_dir)), "speedyspeech")
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)
pwg_config = inference.Config( pwg_config = inference.Config(str(Path(args.inference_dir)), "pwg")
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)

Write Preview
Loading…
Cancel
Save