deepgeek/CosyVoice
2
0
Fork 0
mirror of https://github.com/FunAudioLLM/CosyVoice.git synced 2026-02-05 18:09:24 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix fm train bug

Browse Source
This commit is contained in:
lyuxiang.lx
2026-01-19 10:48:24 +08:00
parent 1dcc59676f
commit 1822c5c908
1 changed files with 1 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
cosyvoice/flow/flow_matching.py
View File

@@ -174,8 +174,7 @@ class ConditionalCFM(BASECFM):
# random timestep # random timestep
t = torch.rand([b, 1, 1], device=mu.device, dtype=mu.dtype) t = torch.rand([b, 1, 1], device=mu.device, dtype=mu.dtype)
if self.t_scheduler == 'cosine':
t = 1 - torch.cos(t * 0.5 * torch.pi)
# sample noise p(x_0) # sample noise p(x_0)
z = torch.randn_like(x1) z = torch.randn_like(x1)