Sam Khoze
37 lines
1.4 KiB
Python
37 lines
1.4 KiB
Python
import os
|
|
import torch
|
|
import argparse
|
|
import sys
|
|
from sys import platform
|
|
|
|
if platform == "win32":
|
|
sys.path.append('./')
|
|
|
|
from TTS.api import TTS
|
|
|
|
def main():
|
|
parser = argparse.ArgumentParser(description="Run TTS with specified parameters.")
|
|
|
|
parser.add_argument('--model', type=str, default="tts_models/multilingual/multi-dataset/xtts_v2", help="The TTS model to use.")
|
|
parser.add_argument('--text', type=str, required=True, help="The text to be converted to speech.")
|
|
parser.add_argument('--speaker_wav', type=str, required=True, help="The path to the speaker's wav file for voice cloning.")
|
|
parser.add_argument('--language', type=str, default="en", help="The language of the text.")
|
|
parser.add_argument('--output_file', type=str, default="output.wav", help="The output file path for the synthesized speech.")
|
|
parser.add_argument('--device', type=str, choices=["cpu", "mps","cuda"], default="cpu" if torch.cuda.is_available() else "cpu", help="The device to run the model on.")
|
|
|
|
args = parser.parse_args()
|
|
|
|
if args.device == "cuda":
|
|
device="cuda"
|
|
else:
|
|
device="cpu"
|
|
|
|
# Init TTS
|
|
tts = TTS(model_path=args.model,config_path=os.path.join(args.model,"config.json")).to(device)
|
|
|
|
# Run TTS and save to file
|
|
tts.tts_to_file(text=args.text, speaker_wav=args.speaker_wav, language=args.language, file_path=args.output_file)
|
|
|
|
if __name__ == "__main__":
|
|
main()
|