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/flow_matching.py
View File

@@ -46,7 +46,7 @@ class BASECFM(torch.nn.Module, ABC):
Returns:
sample: generated mel-spectrogram
shape: (batch_size, n_feats, mel_timesteps)
shape: (batch_size, n_feats, mel_timesteps)
"""
z = torch.randn_like(mu) * temperature
t_span = torch.linspace(0, 1, n_timesteps + 1, device=mu.device)