Fabio
c36510c21e
Some checks failed
Linux Build / Linux cpu (push) Has been cancelled
Linux Build / Linux cpu-clang (push) Has been cancelled
Linux Build / Linux vulkan (push) Has been cancelled
mac-build / macOS CPU (push) Has been cancelled
Nix Build / Nix (cpu) (push) Has been cancelled
Nix Build / Nix (vulkan) (push) Has been cancelled
windows-build / Windows CPU (push) Has been cancelled
69 lines
1.9 KiB
Python
69 lines
1.9 KiB
Python
#!/usr/bin/env python3
|
|
"""Launch the upstream GLM-TTS pipeline on platforms without WeText/Pynini.
|
|
|
|
The Windows PyPI index does not provide the Pynini wheel required by
|
|
WeTextProcessing. For parity inputs that contain no numbers or abbreviations,
|
|
an identity normalizer is sufficient and leaves the model path unchanged.
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import argparse
|
|
import os
|
|
import runpy
|
|
import sys
|
|
import types
|
|
from pathlib import Path
|
|
|
|
import torch
|
|
|
|
|
|
class IdentityNormalizer:
|
|
def __init__(self, *args, **kwargs):
|
|
del args, kwargs
|
|
|
|
def normalize(self, text: str) -> str:
|
|
return text
|
|
|
|
|
|
def install_wetext_identity_stub() -> None:
|
|
modules = {
|
|
"tn": types.ModuleType("tn"),
|
|
"tn.chinese": types.ModuleType("tn.chinese"),
|
|
"tn.chinese.normalizer": types.ModuleType("tn.chinese.normalizer"),
|
|
"tn.english": types.ModuleType("tn.english"),
|
|
"tn.english.normalizer": types.ModuleType("tn.english.normalizer"),
|
|
}
|
|
modules["tn.chinese.normalizer"].Normalizer = IdentityNormalizer
|
|
modules["tn.english.normalizer"].Normalizer = IdentityNormalizer
|
|
sys.modules.update(modules)
|
|
|
|
|
|
def main() -> None:
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument("reference_root", type=Path)
|
|
parser.add_argument("--data", required=True)
|
|
parser.add_argument("--exp-name", default="_audio_cpp_parity")
|
|
args = parser.parse_args()
|
|
|
|
root = args.reference_root.resolve()
|
|
entry = root / "glmtts_inference.py"
|
|
if not entry.is_file():
|
|
raise FileNotFoundError(entry)
|
|
install_wetext_identity_stub()
|
|
if not hasattr(torch, "npu"):
|
|
torch.npu = types.SimpleNamespace(is_available=lambda: False)
|
|
os.chdir(root)
|
|
sys.path.insert(0, str(root))
|
|
sys.argv = [
|
|
str(entry),
|
|
"--data",
|
|
args.data,
|
|
"--exp_name",
|
|
args.exp_name,
|
|
]
|
|
runpy.run_path(str(entry), run_name="__main__")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|