Fix download progress (#78)

2024-05-28 23:05:37 +02:00
parent 546ce10c32
commit 65f4968e5f
1 changed files with 2 additions and 2 deletions
--- a/Libraries/LLM/Load.swift
+++ b/Libraries/LLM/Load.swift
@@ -24,9 +24,9 @@ public func load(

        switch configuration.id {
        case .id(let id):
-            // download the model weights and config
+            // download the model weights
            let repo = Hub.Repo(id: id)
-            let modelFiles = ["config.json", "*.safetensors"]
+            let modelFiles = ["*.safetensors"]
            modelDirectory = try await hub.snapshot(
                from: repo, matching: modelFiles, progressHandler: progressHandler)