Merge pull request #527 from Jerrister/1126

Fix sample timesteps
SWivid · Nov 26, 2024 · d8e622f · d8e622f
2 parents d81911b + 5aa4766
commit d8e622f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/f5_tts/model/cfm.py b/src/f5_tts/model/cfm.py
@@ -193,7 +193,7 @@ def fn(t, x):
             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)