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

R0402

Browse Source
This commit is contained in:
Jim O'Regan
2024-10-02 13:21:00 +00:00
parent 362ba2dce7
commit bd058a68f7
5 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
matcha/hifigan/models.py
View File

@@ -1,7 +1,7 @@
""" from https://github.com/jik876/hifi-gan """ """ from https://github.com/jik876/hifi-gan """
import torch import torch
import torch.nn as nn from torch import nn
import torch.nn.functional as F import torch.nn.functional as F
from torch.nn import AvgPool1d, Conv1d, Conv2d, ConvTranspose1d from torch.nn import AvgPool1d, Conv1d, Conv2d, ConvTranspose1d
from torch.nn.utils import remove_weight_norm, spectral_norm, weight_norm from torch.nn.utils import remove_weight_norm, spectral_norm, weight_norm

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

@@ -2,7 +2,7 @@ import math
from typing import Optional from typing import Optional
import torch import torch
import torch.nn as nn from torch import nn
import torch.nn.functional as F import torch.nn.functional as F
from conformer import ConformerBlock from conformer import ConformerBlock
from diffusers.models.activations import get_activation from diffusers.models.activations import get_activation

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

@@ -3,10 +3,10 @@
import math import math
import torch import torch
import torch.nn as nn from torch import nn
from einops import rearrange from einops import rearrange
import matcha.utils as utils from matcha import utils
from matcha.utils.model import sequence_mask from matcha.utils.model import sequence_mask
log = utils.get_pylogger(__name__) log = utils.get_pylogger(__name__)

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

@@ -1,7 +1,7 @@
from typing import Any, Dict, Optional from typing import Any, Dict, Optional
import torch import torch
import torch.nn as nn from torch import nn
from diffusers.models.attention import ( from diffusers.models.attention import (
GEGLU, GEGLU,
GELU, GELU,

2
matcha/models/matcha_tts.py
View File

@@ -4,7 +4,7 @@ import random
import torch import torch
import matcha.utils.monotonic_align as monotonic_align from matcha.utils import monotonic_align
from matcha import utils from matcha import utils
from matcha.models.baselightningmodule import BaseLightningClass from matcha.models.baselightningmodule import BaseLightningClass
from matcha.models.components.flow_matching import CFM from matcha.models.components.flow_matching import CFM