MLX Swift 示例
MLX Swift 示例程序。
-
MNISTTrainer:一个在 iOS 和 macOS 上运行的示例,可下载 MNIST 训练数据并训练一个 LeNet。
-
LLMEval:一个在 iOS 和 macOS 上运行的示例,可从 Hugging Face 下载 LLM 和分词器,并根据给定的提示生成文本。
-
LinearModelTraining:一个训练简单线性模型的示例。
-
llm-tool:一个命令行工具,用于使用 Hugging Face hub 上提供的各种 LLM 生成文本。
-
mnist-tool:一个命令行工具,用于在 MNIST 数据集上训练 LeNet。
运行
应用程序和命令行工具示例可以通过 Xcode 或命令行运行:
./mlx-run llm-tool --prompt "swift programming language"
另请参阅:
MLXLLM 和 MLXMNIST 库的安装
示例仓库中的 MLXLLM 和 MLXMNIST 库可作为 Swift 包使用。
在你的 Package.swift 中添加以下依赖:
.package(url: "https://github.com/ml-explore/mlx-swift-examples/", branch: "main"),
然后将一个或两个库添加到目标的依赖项中。
.target(
name: "YourTargetName",
dependencies: [
.product(name: "LLM", package: "mlx-swift-examples")
]),
或者,在 Xcode 中将 https://github.com/ml-explore/mlx-swift-examples/
添加到 项目依赖项
,并将 依赖规则
设置为 分支
和 main
。