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

initial commit

Browse Source
This commit is contained in:
David Koski
2024-02-22 10:41:02 -08:00
commit b6d1e14465
29 changed files with 3856 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
Libraries/LLM/LLMModel.swift Normal file
View File

@@ -0,0 +1,12 @@
// Copyright © 2024 Apple Inc.
import Foundation
import MLX
import MLXNN
// Interface for all LLM Models
public protocol LLMModel: Module {
func callAsFunction(_ inputs: MLXArray, cache: [(MLXArray, MLXArray)]?) -> (
MLXArray, [(MLXArray, MLXArray)]
)
}