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

fix inference

Browse Source
This commit is contained in:
Fangjun Kuang 2024-10-28 19:24:09 +08:00
parent ed569a938a
commit ba4df19224
1 changed files with 10 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
egs/ljspeech/TTS/matcha/inference.py
View File

@ -13,8 +13,6 @@ from matcha.hifigan.config import v1, v2, v3
from matcha.hifigan.denoiser import Denoiser
from tokenizer import Tokenizer
from matcha.hifigan.models import Generator as HiFiGAN
from matcha.text import sequence_to_text, text_to_sequence
from matcha.utils.utils import intersperse
from tqdm.auto import tqdm
from train import get_model, get_params
@ -151,8 +149,13 @@ def main():
denoiser = Denoiser(vocoder, mode="zeros")
texts = [
"The Secret Service believed that it was very doubtful that any President would ride regularly in a vehicle with a fixed top, even though transparent.",
"Today as always, men fall into two groups: slaves and free men. Whoever does not have two-thirds of his day for himself, is a slave, whatever he may be: a statesman, a businessman, an official, or a scholar.",
"The Secret Service believed that it was very doubtful that any "
"President would ride regularly in a vehicle with a fixed top, even "
"though transparent.",
"Today as always, men fall into two groups: slaves and free men. "
"Whoever does not have two-thirds of his day for himself, is a slave, "
"whatever he may be: a statesman, a businessman, an official, or a "
"scholar.",
]
# Number of ODE Solver steps
@ -164,7 +167,7 @@ def main():
# Sampling temperature
temperature = 0.667
outputs, rtfs = [], []
rtfs = []
rtfs_w = []
for i, text in enumerate(tqdm(texts)):
output = synthesise(
@ -202,7 +205,8 @@ def main():
print(f"Number of ODE steps: {n_timesteps}")
print(f"Mean RTF:\t\t\t\t{np.mean(rtfs):.6f} ± {np.std(rtfs):.6f}")
print(
f"Mean RTF Waveform (incl. vocoder):\t{np.mean(rtfs_w):.6f} ± {np.std(rtfs_w):.6f}"
"Mean RTF Waveform "
f"(incl. vocoder):\t{np.mean(rtfs_w):.6f} ± {np.std(rtfs_w):.6f}"
)