Skip to content

Commit c470c7b

Browse files
committed
more compile fixes
1 parent 977db17 commit c470c7b

File tree

2 files changed

+8
-2
lines changed

2 files changed

+8
-2
lines changed

Libraries/MLXLLM/LLMModelFactory.swift

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -456,7 +456,10 @@ private struct LLMUserInputProcessor: UserInputProcessor {
456456
/// let modelContainer = try await LLMModelFactory.shared.loadContainer(
457457
/// configuration: LLMRegistry.llama3_8B_4bit)
458458
/// ```
459-
public final class LLMModelFactory: ModelFactory {
459+
public final class LLMModelFactory: GenericModelFactory {
460+
461+
public typealias ContextType = ModelContext
462+
public typealias ContainerType = ModelContainer
460463

461464
public init(
462465
typeRegistry: ModelTypeRegistry<LanguageModel>, modelRegistry: AbstractModelRegistry

Libraries/MLXVLM/VLMModelFactory.swift

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -257,7 +257,10 @@ public typealias ModelRegistry = VLMRegistry
257257
/// let modelContainer = try await VLMModelFactory.shared.loadContainer(
258258
/// configuration: VLMRegistry.paligemma3bMix4488bit)
259259
/// ```
260-
public final class VLMModelFactory: ModelFactory {
260+
public final class VLMModelFactory: GenericModelFactory {
261+
262+
public typealias ContextType = ModelContext
263+
public typealias ContainerType = ModelContainer
261264

262265
public init(
263266
typeRegistry: ModelTypeRegistry<LanguageModel>, processorRegistry: ProcessorTypeRegistry,

0 commit comments

Comments
 (0)