mirror of https://github.com/coqui-ai/TTS.git
Ruff autofix unused imports and import order
This commit is contained in:
parent
4f859824ac
commit
90991e89b4
|
@ -6,10 +6,10 @@ from typing import Union
|
||||||
import numpy as np
|
import numpy as np
|
||||||
from torch import nn
|
from torch import nn
|
||||||
|
|
||||||
|
from TTS.config import load_config
|
||||||
from TTS.utils.audio.numpy_transforms import save_wav
|
from TTS.utils.audio.numpy_transforms import save_wav
|
||||||
from TTS.utils.manage import ModelManager
|
from TTS.utils.manage import ModelManager
|
||||||
from TTS.utils.synthesizer import Synthesizer
|
from TTS.utils.synthesizer import Synthesizer
|
||||||
from TTS.config import load_config
|
|
||||||
|
|
||||||
|
|
||||||
class TTS(nn.Module):
|
class TTS(nn.Module):
|
||||||
|
|
|
@ -1,4 +1,5 @@
|
||||||
"""Get detailed info about the working environment."""
|
"""Get detailed info about the working environment."""
|
||||||
|
import json
|
||||||
import os
|
import os
|
||||||
import platform
|
import platform
|
||||||
import sys
|
import sys
|
||||||
|
@ -6,11 +7,10 @@ import sys
|
||||||
import numpy
|
import numpy
|
||||||
import torch
|
import torch
|
||||||
|
|
||||||
sys.path += [os.path.abspath(".."), os.path.abspath(".")]
|
|
||||||
import json
|
|
||||||
|
|
||||||
import TTS
|
import TTS
|
||||||
|
|
||||||
|
sys.path += [os.path.abspath(".."), os.path.abspath(".")]
|
||||||
|
|
||||||
|
|
||||||
def system_info():
|
def system_info():
|
||||||
return {
|
return {
|
||||||
|
|
|
@ -1,23 +1,17 @@
|
||||||
import os
|
|
||||||
import gc
|
import gc
|
||||||
import torchaudio
|
import os
|
||||||
|
|
||||||
import pandas
|
import pandas
|
||||||
from faster_whisper import WhisperModel
|
|
||||||
from glob import glob
|
|
||||||
|
|
||||||
from tqdm import tqdm
|
|
||||||
|
|
||||||
import torch
|
import torch
|
||||||
import torchaudio
|
import torchaudio
|
||||||
# torch.set_num_threads(1)
|
from faster_whisper import WhisperModel
|
||||||
|
from tqdm import tqdm
|
||||||
|
|
||||||
|
# torch.set_num_threads(1)
|
||||||
from TTS.tts.layers.xtts.tokenizer import multilingual_cleaners
|
from TTS.tts.layers.xtts.tokenizer import multilingual_cleaners
|
||||||
|
|
||||||
torch.set_num_threads(16)
|
torch.set_num_threads(16)
|
||||||
|
|
||||||
|
|
||||||
import os
|
|
||||||
|
|
||||||
audio_types = (".wav", ".mp3", ".flac")
|
audio_types = (".wav", ".mp3", ".flac")
|
||||||
|
|
||||||
|
|
||||||
|
|
|
@ -1,5 +1,5 @@
|
||||||
import os
|
|
||||||
import gc
|
import gc
|
||||||
|
import os
|
||||||
|
|
||||||
from trainer import Trainer, TrainerArgs
|
from trainer import Trainer, TrainerArgs
|
||||||
|
|
||||||
|
|
|
@ -1,19 +1,16 @@
|
||||||
import argparse
|
import argparse
|
||||||
|
import logging
|
||||||
import os
|
import os
|
||||||
import sys
|
import sys
|
||||||
import tempfile
|
import tempfile
|
||||||
|
import traceback
|
||||||
|
|
||||||
import gradio as gr
|
import gradio as gr
|
||||||
import librosa.display
|
|
||||||
import numpy as np
|
|
||||||
|
|
||||||
import os
|
|
||||||
import torch
|
import torch
|
||||||
import torchaudio
|
import torchaudio
|
||||||
import traceback
|
|
||||||
from TTS.demos.xtts_ft_demo.utils.formatter import format_audio_list
|
from TTS.demos.xtts_ft_demo.utils.formatter import format_audio_list
|
||||||
from TTS.demos.xtts_ft_demo.utils.gpt_train import train_gpt
|
from TTS.demos.xtts_ft_demo.utils.gpt_train import train_gpt
|
||||||
|
|
||||||
from TTS.tts.configs.xtts_config import XttsConfig
|
from TTS.tts.configs.xtts_config import XttsConfig
|
||||||
from TTS.tts.models.xtts import Xtts
|
from TTS.tts.models.xtts import Xtts
|
||||||
|
|
||||||
|
@ -91,7 +88,7 @@ sys.stderr = sys.stdout
|
||||||
|
|
||||||
|
|
||||||
# logging.basicConfig(stream=sys.stdout, level=logging.INFO)
|
# logging.basicConfig(stream=sys.stdout, level=logging.INFO)
|
||||||
import logging
|
|
||||||
logging.basicConfig(
|
logging.basicConfig(
|
||||||
level=logging.INFO,
|
level=logging.INFO,
|
||||||
format="%(asctime)s [%(levelname)s] %(message)s",
|
format="%(asctime)s [%(levelname)s] %(message)s",
|
||||||
|
|
|
@ -1,4 +1,4 @@
|
||||||
from dataclasses import asdict, dataclass
|
from dataclasses import dataclass
|
||||||
|
|
||||||
from TTS.encoder.configs.base_encoder_config import BaseEncoderConfig
|
from TTS.encoder.configs.base_encoder_config import BaseEncoderConfig
|
||||||
|
|
||||||
|
|
|
@ -1,4 +1,4 @@
|
||||||
from dataclasses import asdict, dataclass
|
from dataclasses import dataclass
|
||||||
|
|
||||||
from TTS.encoder.configs.base_encoder_config import BaseEncoderConfig
|
from TTS.encoder.configs.base_encoder_config import BaseEncoderConfig
|
||||||
|
|
||||||
|
|
|
@ -7,8 +7,6 @@ License: MIT
|
||||||
|
|
||||||
# Modified code from https://github.com/lucidrains/audiolm-pytorch/blob/main/audiolm_pytorch/hubert_kmeans.py
|
# Modified code from https://github.com/lucidrains/audiolm-pytorch/blob/main/audiolm_pytorch/hubert_kmeans.py
|
||||||
|
|
||||||
import logging
|
|
||||||
from pathlib import Path
|
|
||||||
|
|
||||||
import torch
|
import torch
|
||||||
from einops import pack, unpack
|
from einops import pack, unpack
|
||||||
|
|
|
@ -1,6 +1,5 @@
|
||||||
import functools
|
import functools
|
||||||
import math
|
import math
|
||||||
import os
|
|
||||||
|
|
||||||
import fsspec
|
import fsspec
|
||||||
import torch
|
import torch
|
||||||
|
|
|
@ -2,7 +2,7 @@ import torch
|
||||||
from torch import nn
|
from torch import nn
|
||||||
from torch.nn.modules.conv import Conv1d
|
from torch.nn.modules.conv import Conv1d
|
||||||
|
|
||||||
from TTS.vocoder.models.hifigan_discriminator import DiscriminatorP, MultiPeriodDiscriminator
|
from TTS.vocoder.models.hifigan_discriminator import DiscriminatorP
|
||||||
|
|
||||||
|
|
||||||
class DiscriminatorS(torch.nn.Module):
|
class DiscriminatorS(torch.nn.Module):
|
||||||
|
|
|
@ -1,7 +1,6 @@
|
||||||
# ported from: https://github.com/neonbjb/tortoise-tts
|
# ported from: https://github.com/neonbjb/tortoise-tts
|
||||||
|
|
||||||
import functools
|
import functools
|
||||||
import math
|
|
||||||
import random
|
import random
|
||||||
|
|
||||||
import torch
|
import torch
|
||||||
|
|
|
@ -1,5 +1,3 @@
|
||||||
import math
|
|
||||||
|
|
||||||
import torch
|
import torch
|
||||||
from torch import nn
|
from torch import nn
|
||||||
from transformers import GPT2PreTrainedModel
|
from transformers import GPT2PreTrainedModel
|
||||||
|
|
|
@ -1,4 +1,3 @@
|
||||||
import os
|
|
||||||
import random
|
import random
|
||||||
import sys
|
import sys
|
||||||
|
|
||||||
|
|
|
@ -5,7 +5,6 @@ import torch
|
||||||
import torch.nn as nn
|
import torch.nn as nn
|
||||||
import torchaudio
|
import torchaudio
|
||||||
from coqpit import Coqpit
|
from coqpit import Coqpit
|
||||||
from torch.nn import functional as F
|
|
||||||
from torch.utils.data import DataLoader
|
from torch.utils.data import DataLoader
|
||||||
from trainer.torch import DistributedSampler
|
from trainer.torch import DistributedSampler
|
||||||
from trainer.trainer_utils import get_optimizer, get_scheduler
|
from trainer.trainer_utils import get_optimizer, get_scheduler
|
||||||
|
|
|
@ -4,13 +4,11 @@
|
||||||
|
|
||||||
import argparse
|
import argparse
|
||||||
import csv
|
import csv
|
||||||
import os
|
|
||||||
import re
|
import re
|
||||||
import string
|
import string
|
||||||
import sys
|
import sys
|
||||||
|
|
||||||
# fmt: off
|
# fmt: off
|
||||||
|
|
||||||
# ================================================================================ #
|
# ================================================================================ #
|
||||||
# basic constant
|
# basic constant
|
||||||
# ================================================================================ #
|
# ================================================================================ #
|
||||||
|
|
|
@ -14,7 +14,7 @@ from TTS.model import BaseTrainerModel
|
||||||
from TTS.tts.datasets.dataset import TTSDataset
|
from TTS.tts.datasets.dataset import TTSDataset
|
||||||
from TTS.tts.utils.data import get_length_balancer_weights
|
from TTS.tts.utils.data import get_length_balancer_weights
|
||||||
from TTS.tts.utils.languages import LanguageManager, get_language_balancer_weights
|
from TTS.tts.utils.languages import LanguageManager, get_language_balancer_weights
|
||||||
from TTS.tts.utils.speakers import SpeakerManager, get_speaker_balancer_weights, get_speaker_manager
|
from TTS.tts.utils.speakers import SpeakerManager, get_speaker_balancer_weights
|
||||||
from TTS.tts.utils.synthesis import synthesis
|
from TTS.tts.utils.synthesis import synthesis
|
||||||
from TTS.tts.utils.visual import plot_alignment, plot_spectrogram
|
from TTS.tts.utils.visual import plot_alignment, plot_spectrogram
|
||||||
|
|
||||||
|
|
|
@ -1,7 +1,5 @@
|
||||||
from dataclasses import asdict, dataclass, field
|
from dataclasses import dataclass, field
|
||||||
from typing import Dict, List
|
from typing import List
|
||||||
|
|
||||||
from coqpit import Coqpit, check_argument
|
|
||||||
|
|
||||||
from TTS.config import BaseAudioConfig, BaseDatasetConfig, BaseTrainingConfig
|
from TTS.config import BaseAudioConfig, BaseDatasetConfig, BaseTrainingConfig
|
||||||
|
|
||||||
|
|
|
@ -1,8 +1,6 @@
|
||||||
import math
|
import math
|
||||||
|
|
||||||
import numpy as np
|
|
||||||
import torch
|
import torch
|
||||||
from torch import nn
|
|
||||||
from torch.nn import functional as F
|
from torch.nn import functional as F
|
||||||
|
|
||||||
|
|
||||||
|
|
|
@ -1,11 +1,9 @@
|
||||||
import struct
|
|
||||||
from pathlib import Path
|
from pathlib import Path
|
||||||
from typing import Optional, Union
|
from typing import Optional, Union
|
||||||
|
|
||||||
# import webrtcvad
|
# import webrtcvad
|
||||||
import librosa
|
import librosa
|
||||||
import numpy as np
|
import numpy as np
|
||||||
from scipy.ndimage.morphology import binary_dilation
|
|
||||||
|
|
||||||
from TTS.vc.modules.freevc.speaker_encoder.hparams import (
|
from TTS.vc.modules.freevc.speaker_encoder.hparams import (
|
||||||
audio_norm_target_dBFS,
|
audio_norm_target_dBFS,
|
||||||
|
|
|
@ -1,4 +1,3 @@
|
||||||
from pathlib import Path
|
|
||||||
from time import perf_counter as timer
|
from time import perf_counter as timer
|
||||||
from typing import List, Union
|
from typing import List, Union
|
||||||
|
|
||||||
|
|
|
@ -1,6 +1,3 @@
|
||||||
import os
|
|
||||||
|
|
||||||
from coqpit import Coqpit
|
|
||||||
from trainer import Trainer, TrainerArgs
|
from trainer import Trainer, TrainerArgs
|
||||||
|
|
||||||
from TTS.tts.configs.shared_configs import BaseAudioConfig
|
from TTS.tts.configs.shared_configs import BaseAudioConfig
|
||||||
|
|
|
@ -4,7 +4,6 @@ import torch
|
||||||
from trainer import Trainer, TrainerArgs
|
from trainer import Trainer, TrainerArgs
|
||||||
|
|
||||||
from TTS.bin.compute_embeddings import compute_embeddings
|
from TTS.bin.compute_embeddings import compute_embeddings
|
||||||
from TTS.bin.resample import resample_files
|
|
||||||
from TTS.config.shared_configs import BaseDatasetConfig
|
from TTS.config.shared_configs import BaseDatasetConfig
|
||||||
from TTS.tts.configs.vits_config import VitsConfig
|
from TTS.tts.configs.vits_config import VitsConfig
|
||||||
from TTS.tts.datasets import load_tts_samples
|
from TTS.tts.datasets import load_tts_samples
|
||||||
|
|
2
setup.py
2
setup.py
|
@ -23,12 +23,12 @@
|
||||||
import os
|
import os
|
||||||
import subprocess
|
import subprocess
|
||||||
import sys
|
import sys
|
||||||
from packaging.version import Version
|
|
||||||
|
|
||||||
import numpy
|
import numpy
|
||||||
import setuptools.command.build_py
|
import setuptools.command.build_py
|
||||||
import setuptools.command.develop
|
import setuptools.command.develop
|
||||||
from Cython.Build import cythonize
|
from Cython.Build import cythonize
|
||||||
|
from packaging.version import Version
|
||||||
from setuptools import Extension, find_packages, setup
|
from setuptools import Extension, find_packages, setup
|
||||||
|
|
||||||
python_version = sys.version.split()[0]
|
python_version = sys.version.split()[0]
|
||||||
|
|
|
@ -6,10 +6,10 @@ import numpy as np
|
||||||
import torch
|
import torch
|
||||||
from torch.utils.data import DataLoader
|
from torch.utils.data import DataLoader
|
||||||
|
|
||||||
from TTS.tts.datasets.dataset import TTSDataset
|
|
||||||
from tests import get_tests_data_path, get_tests_output_path
|
from tests import get_tests_data_path, get_tests_output_path
|
||||||
from TTS.tts.configs.shared_configs import BaseDatasetConfig, BaseTTSConfig
|
from TTS.tts.configs.shared_configs import BaseDatasetConfig, BaseTTSConfig
|
||||||
from TTS.tts.datasets import load_tts_samples
|
from TTS.tts.datasets import load_tts_samples
|
||||||
|
from TTS.tts.datasets.dataset import TTSDataset
|
||||||
from TTS.tts.utils.text.tokenizer import TTSTokenizer
|
from TTS.tts.utils.text.tokenizer import TTSTokenizer
|
||||||
from TTS.utils.audio import AudioProcessor
|
from TTS.utils.audio import AudioProcessor
|
||||||
|
|
||||||
|
|
|
@ -4,8 +4,7 @@ import unittest
|
||||||
import torch
|
import torch
|
||||||
|
|
||||||
from tests import get_tests_input_path
|
from tests import get_tests_input_path
|
||||||
from TTS.vc.configs.freevc_config import FreeVCConfig
|
from TTS.vc.models.freevc import FreeVC, FreeVCConfig
|
||||||
from TTS.vc.models.freevc import FreeVC
|
|
||||||
|
|
||||||
# pylint: disable=unused-variable
|
# pylint: disable=unused-variable
|
||||||
# pylint: disable=no-self-use
|
# pylint: disable=no-self-use
|
||||||
|
|
Loading…
Reference in New Issue