deepgeek/Matcha-TTS
2
0
Fork 0
mirror of https://github.com/shivammehta25/Matcha-TTS.git synced 2026-02-04 17:59:19 +08:00
Code Issues Packages Projects Releases Wiki Activity

disable consider-using-from-import instead

Browse Source
This commit is contained in:
Jim O'Regan
2024-10-02 14:29:06 +00:00
parent a220f283e3
commit b0ba920dc1
5 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
matcha/hifigan/models.py
View File

@@ -2,7 +2,7 @@
import torch
import torch.nn.functional as F
from torch import nn
import torch.nn as nn # pylint: disable=consider-using-from-import
from torch.nn import AvgPool1d, Conv1d, Conv2d, ConvTranspose1d
from torch.nn.utils import remove_weight_norm, spectral_norm, weight_norm

2
matcha/models/components/decoder.py
View File

@@ -2,11 +2,11 @@ import math
from typing import Optional
import torch
import torch.nn as nn # pylint: disable=consider-using-from-import
import torch.nn.functional as F
from conformer import ConformerBlock
from diffusers.models.activations import get_activation
from einops import pack, rearrange, repeat
from torch import nn
from matcha.models.components.transformer import BasicTransformerBlock

2
matcha/models/components/text_encoder.py
View File

@@ -3,8 +3,8 @@
import math
import torch
import torch.nn as nn # pylint: disable=consider-using-from-import
from einops import rearrange
from torch import nn
from matcha import utils
from matcha.utils.model import sequence_mask

2
matcha/models/components/transformer.py
View File

@@ -1,6 +1,7 @@
from typing import Any, Dict, Optional
import torch
import torch.nn as nn # pylint: disable=consider-using-from-import
from diffusers.models.attention import (
GEGLU,
GELU,
@@ -11,7 +12,6 @@ from diffusers.models.attention import (
from diffusers.models.attention_processor import Attention
from diffusers.models.lora import LoRACompatibleLinear
from diffusers.utils.torch_utils import maybe_allow_in_graph
from torch import nn
class SnakeBeta(nn.Module):

2
matcha/models/matcha_tts.py
View File

@@ -4,11 +4,11 @@ import random
import torch
import matcha.utils.monotonic_align as monotonic_align # pylint: disable=consider-using-from-import
from matcha import utils
from matcha.models.baselightningmodule import BaseLightningClass
from matcha.models.components.flow_matching import CFM
from matcha.models.components.text_encoder import TextEncoder
from matcha.utils import monotonic_align
from matcha.utils.model import (
denormalize,
duration_loss,