From 39f8e2c6a0bd3bb57b1779ee0fc9af300a195b7d Mon Sep 17 00:00:00 2001 From: manmay-nakhashi Date: Mon, 10 Jul 2023 00:36:22 +0530 Subject: [PATCH] revert deepspeed version due to crash in do_tts --- requirements.txt | 2 +- tortoise/do_tts.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/requirements.txt b/requirements.txt index 0cb43cb..ac40d50 100644 --- a/requirements.txt +++ b/requirements.txt @@ -18,4 +18,4 @@ appdirs nbconvert==5.3.1 tornado==4.2 pydantic==1.9.0 -deepspeed=9.0.0 +deepspeed==0.8.3 diff --git a/tortoise/do_tts.py b/tortoise/do_tts.py index 522afa0..3c0a481 100644 --- a/tortoise/do_tts.py +++ b/tortoise/do_tts.py @@ -13,6 +13,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='fast') + parser.add_argument('--use_deepspeed', type=str, help='Which voice preset to use.', default=False) 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) @@ -23,8 +24,7 @@ if __name__ == '__main__': 'Increasing this can in some cases reduce the likelihood of multiple speakers. Defaults to 0 (disabled)', default=.0) args = parser.parse_args() os.makedirs(args.output_path, exist_ok=True) - - tts = TextToSpeech(models_dir=args.model_dir) + 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):