fix style and tests

This commit is contained in:
manmay-nakhashi 2023-05-02 01:05:22 +05:30
parent 49feaf5fa1
commit 771bff8c1f
3 changed files with 7 additions and 5 deletions

View File

@ -337,9 +337,6 @@ If you don't specify any models, then it uses LJSpeech based English model.
if args.model_name is not None and not args.model_path:
model_path, config_path, model_item = manager.download_model(args.model_name)
# tortoise model
if model_path.split("--")[-1] == "tortoise-v2":
model_dir = model_path
# tts model
if model_item["model_type"] == "tts_models":
tts_path = model_path
@ -352,6 +349,13 @@ If you don't specify any models, then it uses LJSpeech based English model.
vc_path = model_path
vc_config_path = config_path
# tortoise model
if model_path.split("--")[-1] == "tortoise-v2":
model_dir = model_path
tts_path = None
tts_config_path = None
args.vocoder_name = None
# load vocoder
if args.vocoder_name is not None and not args.vocoder_path:
vocoder_path, vocoder_config_path, _ = manager.download_model(args.vocoder_name)

View File

@ -1,5 +1,4 @@
from dataclasses import dataclass, field
from typing import List
from TTS.tts.configs.shared_configs import BaseTTSConfig
from TTS.tts.models.tortoise import TortoiseArgs, TortoiseAudioConfig

View File

@ -1,4 +1,3 @@
import os
import random
from contextlib import contextmanager
from dataclasses import dataclass