feat: migration to mlx-swift-lm v3
This commit is contained in:
@@ -1,8 +1,10 @@
|
||||
import Foundation
|
||||
import Hub
|
||||
import HuggingFace
|
||||
import MLX
|
||||
import MLXHuggingFace
|
||||
import MLXLMCommon
|
||||
import MLXVLM
|
||||
import Tokenizers
|
||||
import XCTest
|
||||
@testable import MLX_Server
|
||||
|
||||
@@ -230,10 +232,9 @@ private actor LocalGemmaFixture {
|
||||
}
|
||||
|
||||
let loadTask = Task<ModelContainer, Error> {
|
||||
let cachesDir = FileManager.default.urls(for: .cachesDirectory, in: .userDomainMask).first
|
||||
let hub = HubApi(downloadBase: cachesDir, cache: nil)
|
||||
return try await VLMModelFactory.shared.loadContainer(
|
||||
hub: hub,
|
||||
from: #hubDownloader(HubClient.default),
|
||||
using: #huggingFaceTokenizerLoader(),
|
||||
configuration: ModelConfiguration(directory: localDir),
|
||||
progressHandler: { _ in }
|
||||
)
|
||||
|
||||
Reference in New Issue
Block a user