From 2478dc255e73a9feec6a757532a6f6eff5ad45e1 Mon Sep 17 00:00:00 2001 From: ken11o2 Date: Mon, 4 Sep 2023 19:13:45 +0000 Subject: [PATCH] update TextToSpeech --- tortoise/do_tts.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/tortoise/do_tts.py b/tortoise/do_tts.py index bb18eb2..88cb824 100755 --- a/tortoise/do_tts.py +++ b/tortoise/do_tts.py @@ -14,6 +14,7 @@ if __name__ == '__main__': parser.add_argument('--voice', type=str, help='Selects the voice to use for generation. See options in voices/ directory (and add your own!) ' 'Use the & character to join two voices together. Use a comma to perform inference on multiple voices.', default='random') parser.add_argument('--preset', type=str, help='Which voice preset to use.', default='standard') + parser.add_argument('--use_deepspeed', type=bool, help='Use deepspeed for speed bump.', default=True) parser.add_argument('--output_path', type=str, help='Where to store outputs.', default='results/') parser.add_argument('--model_dir', type=str, help='Where to find pretrained model checkpoints. Tortoise automatically downloads these to .models, so this' 'should only be specified if you have custom checkpoints.', default=MODELS_DIR) @@ -37,8 +38,8 @@ if __name__ == '__main__': os.makedirs(args.output_path, exist_ok=True) - - tts = TextToSpeech(models_dir=args.model_dir) + #print(f'use_deepspeed do_tts_debug {use_deepspeed}') + tts = TextToSpeech(models_dir=args.model_dir, use_deepspeed=args.use_deepspeed) selected_voices = args.voice.split(',') for k, selected_voice in enumerate(selected_voices):