Allow setting a downloadBase so downloaded models are not forced into…

… the user's Documents folder (#34) * Allow setting a downloadBase so downloaded models are not forced into the user's Documents folder * Update Sources/WhisperKit/Core/WhisperKit.swift Co-authored-by: Zach Nagengast <[email protected]> --------- Co-authored-by: Atila Orhon <[email protected]> Co-authored-by: Zach Nagengast <[email protected]>
argmaxinc · Feb 21, 2024 · 8d581f4 · 8d581f4
1 parent 228630c
commit 8d581f4
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/Sources/WhisperKit/Core/WhisperKit.swift b/Sources/WhisperKit/Core/WhisperKit.swift
@@ -152,8 +152,8 @@ public class WhisperKit {
         return sortedModels
     }
 
-    public static func download(variant: String, from repo: String = "argmaxinc/whisperkit-coreml", progressCallback: ((Progress) -> Void)? = nil) async throws -> URL? {
-        let hubApi = HubApi()
+    public static func download(variant: String, downloadBase: URL? = nil, from repo: String = "argmaxinc/whisperkit-coreml", progressCallback: ((Progress) -> Void)? = nil) async throws -> URL? {
+        let hubApi = HubApi(downloadBase: downloadBase)
         let repo = Hub.Repo(id: repo, type: .models)
         do {
             let modelFolder = try await hubApi.snapshot(from: repo, matching: ["*\(variant.description)/*"]) { progress in