苹果自家的框架,专为 M 系列芯片的 Mac 做的。在 Mac 上找模型时,认准名字里带 MLX 的版本。
适合:在自己电脑跑 AI
我的 Mac 是 M5 芯片,本地模型都挑 MLX 格式的,同一个模型比别的格式跑得快。
苹果的机器学习研究团队给自家芯片做了一套框架,叫 MLX。mlx-lm 就是用它跑大语言模型的 Python 工具包:一行命令下载模型、在终端里聊天,还能把模型压成 4bit 这类更小的版本。LM Studio、oMLX 这些 Mac 软件,底下用的也是 MLX。
pip install mlx-lm(pip 是 Python 装软件包的工具)。mlx_lm.chat,它会先下载一个默认的小模型(Llama 3.2 3B 的 4bit 版),然后在终端里跟你聊。--model 和模型在 Hugging Face 上的名字,比如 mlx_lm.chat --model mlx-community/Mistral-7B-Instruct-v0.3-4bit。mlx-community 名下有几千个已经转好的模型,挑 4bit 的最省内存。mlx_lm.generate --prompt "你的问题",它答完就退出。https://github.com/ml-explore/mlx-lm
上面的介绍是我们按 mlx-lm 和 MLX 的 GitHub 说明用自己的话整理的,细节以原作者为准。