File size: 2,421 Bytes
1eac567 e8ff114 1eac567 e8ff114 1eac567 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 | """maya "text" -l hi -v Ahana -o speech.wav [--normalize] [--seed N] [--ref-audio a.wav --ref-text "..."]"""
import argparse
import json
def main(argv=None):
ap = argparse.ArgumentParser(prog="maya", description="Maya TTS on Apple Silicon / CPU")
ap.add_argument("text", nargs="?", help="text to speak")
ap.add_argument("-l", "--lang", default="en", help="as bn en gu hi kn ml mr or pa ta te")
ap.add_argument("-v", "--voice", default=None, help="named voice (see --list-voices); default: plain")
ap.add_argument("-o", "--out", default="speech.wav")
ap.add_argument("--normalize", action="store_true", help="expand numbers, currency, dates, URLs ...")
ap.add_argument("--seed", type=int, default=None)
ap.add_argument("--ref-audio", help="reference clip for zero-shot cloning")
ap.add_argument("--ref-text", help="transcript of the reference clip")
ap.add_argument("--ref-lang", help="language of the reference clip (default: --lang)")
ap.add_argument("--models", default=None, help="dir holding maya2-native/, mimi/, ecapa/ (default: Hugging Face cache)")
ap.add_argument("--device", default=None,
help="mlx | mps | cpu | cuda (default: mlx when installed on Apple Silicon, else best available)")
ap.add_argument("--bits", type=int, choices=(8,), default=None,
help="mlx: 8-bit weights for the backbone and depth decoder (~1.4x faster, ~3%% logit drift)")
ap.add_argument("--list-voices", action="store_true")
a = ap.parse_args(argv)
import soundfile as sf
from maya_mac.engine import SR, MayaTTS
tts = MayaTTS.from_pretrained(a.models, device=a.device, **({"backbone_bits": a.bits, "depth_bits": a.bits}
if a.bits else {}))
print(f"loaded on {tts.device} in {tts.load_s:.1f}s")
if a.list_voices:
print("\n".join(tts.voices))
return
if not a.text:
ap.error("text is required")
wav, info = tts.generate(a.text, lang=a.lang, voice=a.voice, normalize=a.normalize, seed=a.seed,
ref_audio=a.ref_audio, ref_text=a.ref_text, ref_lang=a.ref_lang, return_info=True)
sf.write(a.out, wav, SR)
print(json.dumps({k: (round(v, 3) if isinstance(v, float) else v) for k, v in info.items()}, ensure_ascii=False))
print(f"wrote {a.out}")
if __name__ == "__main__":
main()
|