Skip to content

Commit 65f4968

Browse files
Fix download progress (#78)
1 parent 546ce10 commit 65f4968

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

Libraries/LLM/Load.swift

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -24,9 +24,9 @@ public func load(
2424

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

0 commit comments

Comments
 (0)