Fix duration from predictor.

lucasnewman · Oct 21, 2024 · 3f7614c · 3f7614c
1 parent 46d02d8
commit 3f7614c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/Sources/F5TTS/F5TTS.swift b/Sources/F5TTS/F5TTS.swift
@@ -105,7 +105,7 @@ public class F5TTS: Module {
 
         if resolvedDuration == nil, let durationPredictor = self._durationPredictor {
             let estimatedDurationInSeconds = durationPredictor(cond, text: text).item(Float32.self)
-            resolvedDuration = MLXArray(Int(Double(estimatedDurationInSeconds) * F5TTS.framesPerSecond)) + lens
+            resolvedDuration = MLXArray(Int(Double(estimatedDurationInSeconds) * F5TTS.framesPerSecond))
             print("Generating \(estimatedDurationInSeconds) seconds (\(resolvedDuration) total frames) of audio...")
         }