gitea-mirror/F5-TTS
1
0
Fork 0
mirror of https://github.com/SWivid/F5-TTS.git synced 2026-07-28 13:26:59 -07:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #527 from Jerrister/1126

Browse Source 
Fix sample timesteps
This commit is contained in:
Yushen CHEN
2024-11-26 14:14:19 +08:00
committed by GitHub
parent d81911b0c0 5aa4766033
commit d8e622f004
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/f5_tts/model/cfm.py
+1 -1
View File
@@ -193,7 +193,7 @@ class CFM(nn.Module):
y0 = (1 - t_start) * y0 + t_start * test_cond
steps = int(steps * (1 - t_start))
t = torch.linspace(t_start, 1, steps, device=self.device, dtype=step_cond.dtype)
t = torch.linspace(t_start, 1, steps + 1, device=self.device, dtype=step_cond.dtype)
if sway_sampling_coef is not None:
t = t + sway_sampling_coef * (torch.cos(torch.pi / 2 * t) - 1 + t)