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

Adding docstrings

Browse Source
This commit is contained in:
Shivam Mehta
2023-09-17 06:50:46 +00:00
parent c079e5254a
commit a9251ed984
10 changed files with 47 additions and 32 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -69,6 +69,7 @@ class Downsample1D(nn.Module):
def forward(self, x):
return self.conv(x)
class TimestepEmbedding(nn.Module):
def __init__(
self,
@@ -115,6 +116,7 @@ class TimestepEmbedding(nn.Module):
sample = self.post_act(sample)
return sample
class Upsample1D(nn.Module):
"""A 1D upsampling layer with an optional convolution.