mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-09-07 08:04:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update test_onnx.py

Browse Source
This commit is contained in:
jinzr 2023-12-05 10:45:05 +08:00
parent fefae1f8df
commit 64903abf77
1 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
egs/vctk/TTS/vits/test_onnx.py
View File

@ -28,10 +28,11 @@ Use the onnx model to generate a wav:
import argparse import argparse
import logging import logging
from pathlib import Path
import onnxruntime as ort import onnxruntime as ort
import torch import torch
import torchaudio import torchaudio
from tokenizer import Tokenizer from tokenizer import Tokenizer
@ -47,6 +48,12 @@ def get_parser():
help="Path to the onnx model.", help="Path to the onnx model.",
) )
parser.add_argument(
"--speakers",
type=Path,
default=Path("data/speakers.txt"),
help="Path to speakers.txt file.",
)
parser.add_argument( parser.add_argument(
"--tokens", "--tokens",
type=str, type=str,