update notes, test=doc

3 years ago · e0d222e674
parent f264b912fc
commit e0d222e674
2 changed files with 2 additions and 2 deletions
--- a/paddlespeech/t2s/exps/ort_predict.py
+++ b/paddlespeech/t2s/exps/ort_predict.py
@ -36,7 +36,7 @@ def get_sess(args, filed='am'):
    sess_options.execution_mode = ort.ExecutionMode.ORT_SEQUENTIAL

    if args.device == "gpu":
-        # fastspeech2 can't use trt now!
+        # fastspeech2/mb_melgan can't use trt now!
        if args.use_trt:
            providers = ['TensorrtExecutionProvider']
        else:
--- a/paddlespeech/t2s/exps/ort_predict_e2e.py
+++ b/paddlespeech/t2s/exps/ort_predict_e2e.py
@ -36,7 +36,7 @@ def get_sess(args, filed='am'):
    sess_options.execution_mode = ort.ExecutionMode.ORT_SEQUENTIAL

    if args.device == "gpu":
-        # fastspeech2 can't use trt now!
+        # fastspeech2/mb_melgan can't use trt now!
        if args.use_trt:
            providers = ['TensorrtExecutionProvider']
        else: