zhangshuai/mlx-swift-examples
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix download progress (#78)

Browse Source
This commit is contained in:
Anthony DePasquale
2024-05-28 23:05:37 +02:00
committed by GitHub
parent 546ce10c32
commit 65f4968e5f
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
Libraries/LLM/Load.swift
View File

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