mrq/vall-e
5
9
Fork 0
Code Issues 8 Pull Requests Packages Projects Releases Wiki Activity

apply phoneme cleanup in inferencing as well

Browse Source
This commit is contained in:
mrq 2023-10-10 19:21:19 -05:00
parent b4405c98ea
commit 100dd164e6
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
vall_e/inference.py
View File

@ -13,7 +13,7 @@ from .utils import to_device
from .config import cfg from .config import cfg
from .models import get_models from .models import get_models
from .engines import load_engines, deepspeed_available from .engines import load_engines, deepspeed_available
from .data import get_phone_symmap, _load_quants from .data import get_phone_symmap, _load_quants, _cleanup_phones
if deepspeed_available: if deepspeed_available:
import deepspeed import deepspeed
@ -116,6 +116,7 @@ class TTS():
return text return text
content = g2p.encode(text, language=language) content = g2p.encode(text, language=language)
content = _cleanup_phones( content )
# ick # ick
try: try:
phones = ["<s>"] + [ " " if not p else p for p in content ] + ["</s>"] phones = ["<s>"] + [ " " if not p else p for p in content ] + ["</s>"]