Early stopping checks for CLI

[Zach Nagengast]

Similar to the example app, we should bring over the options for early stopping and add CLI arguments for setting them.

WhisperKit/Examples/WhisperAX/WhisperAX/Views/ContentView.swift

Lines 1040 to 1069 in 0f19f7e

 // Early stopping checks 
 let decodingCallback: ((TranscriptionProgress) -> Bool?) = { progress in 
     DispatchQueue.main.async { 
         let fallbacks = Int(progress.timings.totalDecodingFallbacks) 
         if progress.text.count < currentText.count { 
             if fallbacks == self.currentFallbacks { 
                 self.unconfirmedText.append(currentText) 
             } else { 
                 print("Fallback occured: \(fallbacks)") 
             } 
         } 
         self.currentText = progress.text 
         self.currentFallbacks = fallbacks 
     } 
     // Check early stopping 
     let currentTokens = progress.tokens 
     let checkWindow = Int(compressionCheckWindow) 
     if currentTokens.count > checkWindow { 
         let checkTokens: [Int] = currentTokens.suffix(checkWindow) 
         let compressionRatio = compressionRatio(of: checkTokens) 
         if compressionRatio > options.compressionRatioThreshold! { 
             return false 
         } 
     } 
     if progress.avgLogprob! < options.logProbThreshold! { 
         return false 
     } 
  
     return nil 
 } 

[Atila Orhon]

Indeed, we will aim for parity between the WhisperAX sample app and the CLI before the stable release.