mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-08-14 12:32:20 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update matcha_tts.py

Browse Source
This commit is contained in:
zr_jin 2024-11-04 14:28:10 +08:00
parent 729e86edb9
commit 9c2a18493a
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
egs/ljspeech/TTS/matcha/models/matcha_tts.py
View File

@ -2,17 +2,17 @@ import datetime as dt
import math import math
import random import random
import matcha.monotonic_align as monotonic_align import monotonic_align as monotonic_align
import torch import torch
from matcha.model import ( from model import (
denormalize, denormalize,
duration_loss, duration_loss,
fix_len_compatibility, fix_len_compatibility,
generate_path, generate_path,
sequence_mask, sequence_mask,
) )
from matcha.models.components.flow_matching import CFM from models.components.flow_matching import CFM
from matcha.models.components.text_encoder import TextEncoder from models.components.text_encoder import TextEncoder
class MatchaTTS(torch.nn.Module): # 🍵 class MatchaTTS(torch.nn.Module): # 🍵