Fix duration from predictor.

lucasnewman · lucasnewman · commit 3f7614c80b67 · 2024-10-21T15:04:03.000-07:00
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...")
         }
 

Original file line number	Diff line number	Diff line change
`@@ -105,7 +105,7 @@ public class F5TTS: Module {`
`105`	`105`
`106`	`106`	`if resolvedDuration == nil, let durationPredictor = self._durationPredictor {`
`107`	`107`	`let estimatedDurationInSeconds = durationPredictor(cond, text: text).item(Float32.self)`
`108`		`- resolvedDuration = MLXArray(Int(Double(estimatedDurationInSeconds) * F5TTS.framesPerSecond)) + lens`
	`108`	`+ resolvedDuration = MLXArray(Int(Double(estimatedDurationInSeconds) * F5TTS.framesPerSecond))`
`109`	`109`	`print("Generating \(estimatedDurationInSeconds) seconds (\(resolvedDuration) total frames) of audio...")`
`110`	`110`	`}`
`111`	`111`