Merge pull request #4 from rudrankriyam/main

Pass progress in the completion handler of generate method
lucasnewman · Nov 29, 2024 · 6ae8953 · 6ae8953
2 parents 3f7614c + a9e34c7
commit 6ae8953
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/Sources/F5TTS/F5TTS.swift b/Sources/F5TTS/F5TTS.swift
@@ -240,6 +240,7 @@ public class F5TTS: Module {
             vocoder: vocos.decode
         ) { progress in
             print("Generation progress: \(progress)")
+            progressHandler?(progress)
         }
 
         return outputAudio[audio.shape[0]...]