Formatting

argmaxinc · Jan 16, 2025 · 4f82b35 · 4f82b35
1 parent ee58681
commit 4f82b35
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/Sources/WhisperKit/Core/WhisperKit.swift b/Sources/WhisperKit/Core/WhisperKit.swift
@@ -72,7 +72,6 @@ open class WhisperKit {
             modelFolder: config.modelFolder,
             download: config.download
         )
-
 
         if let prewarm = config.prewarm, prewarm {
             Logging.info("Prewarming models...")
@@ -314,7 +313,7 @@ open class WhisperKit {
                     variant: modelVariant,
                     downloadBase: downloadBase,
                     useBackgroundSession: useBackgroundDownloadSession,
-                    from: repo, 
+                    from: repo,
                     token: modelToken
                 )
             } catch {

diff --git a/Tests/WhisperKitTests/RegressionTests.swift b/Tests/WhisperKitTests/RegressionTests.swift
@@ -84,7 +84,7 @@ class RegressionTests: XCTestCase {
         if let modelSizeEnv = ProcessInfo.processInfo.environment["MODEL_NAME"], !modelSizeEnv.isEmpty {
             modelsToTest = [modelSizeEnv]
             Logging.debug("Model size: \(modelSizeEnv)")
-            
+
             if let repoEnv = ProcessInfo.processInfo.environment["MODEL_REPO"] {
                 modelReposToTest = [repoEnv]
                 Logging.debug("Using repo: \(repoEnv)")
@@ -632,7 +632,7 @@ class RegressionTests: XCTestCase {
         let modelSize = try fileManager.allocatedSizeOfDirectory(at: folder)
         return Double(modelSize / (1024 * 1024)) // Convert to MB
     }
-    
+
     public func initWhisperKitTask(testConfig config: TestConfig, verbose: Bool, logLevel: Logging.LogLevel) -> Task<WhisperKit, Error> {
         // Create the initialization task
         let initializationTask = Task { () -> WhisperKit in