瀏覽代碼

fix fm train bug

lyuxiang.lx 3 月之前
父節點
當前提交
1822c5c908
共有 1 個文件被更改,包括 1 次插入2 次删除
  1. 1 2
      cosyvoice/flow/flow_matching.py

+ 1 - 2
cosyvoice/flow/flow_matching.py

@@ -174,8 +174,7 @@ class ConditionalCFM(BASECFM):
 
         # random timestep
         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)
         z = torch.randn_like(x1)