mirror of https://github.com/coqui-ai/TTS.git
fix style and tests
This commit is contained in:
parent
49feaf5fa1
commit
771bff8c1f
|
@ -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:
|
if args.model_name is not None and not args.model_path:
|
||||||
model_path, config_path, model_item = manager.download_model(args.model_name)
|
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
|
# tts model
|
||||||
if model_item["model_type"] == "tts_models":
|
if model_item["model_type"] == "tts_models":
|
||||||
tts_path = model_path
|
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_path = model_path
|
||||||
vc_config_path = config_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
|
# load vocoder
|
||||||
if args.vocoder_name is not None and not args.vocoder_path:
|
if args.vocoder_name is not None and not args.vocoder_path:
|
||||||
vocoder_path, vocoder_config_path, _ = manager.download_model(args.vocoder_name)
|
vocoder_path, vocoder_config_path, _ = manager.download_model(args.vocoder_name)
|
||||||
|
|
|
@ -1,5 +1,4 @@
|
||||||
from dataclasses import dataclass, field
|
from dataclasses import dataclass, field
|
||||||
from typing import List
|
|
||||||
|
|
||||||
from TTS.tts.configs.shared_configs import BaseTTSConfig
|
from TTS.tts.configs.shared_configs import BaseTTSConfig
|
||||||
from TTS.tts.models.tortoise import TortoiseArgs, TortoiseAudioConfig
|
from TTS.tts.models.tortoise import TortoiseArgs, TortoiseAudioConfig
|
||||||
|
|
|
@ -1,4 +1,3 @@
|
||||||
import os
|
|
||||||
import random
|
import random
|
||||||
from contextlib import contextmanager
|
from contextlib import contextmanager
|
||||||
from dataclasses import dataclass
|
from dataclasses import dataclass
|
||||||
|
|
Loading…
Reference in New Issue