OpenAI联创Karpathy爱上羊驼：纯C代码实现婴儿Llama2，MacBook可运行，已经揽1.6k星

编纂:杜伟

上个周末 ，联创Karpathy 花了全部周六来忙一个 Llama2 的爱上幽默名目。

这一周来，羊驼婴儿Meta 开源的代码 Llama2 火遍了全部 AI 社区。

这不 ，实现连特斯拉前 AI 总监 、可运年初重回 OpenAI 的行已k星 Andrej Karpathy 也坐不住了。他运用周末光阴 ，经揽做了一个对于 Llama2 的联创幽默名目 ——「llama2.c」。

GitHub 地址：https://github.com/karpathy/llama2.c

详细是爱上甚么呢？他展现「llama2.c」可能让你在 PyTorch 中磨炼一个 baby Llama2 模子 ，而后运用近 500 行纯 C、羊驼婴儿无任何依赖性的代码文件妨碍推理。而且，实现这个预磨炼模子可能在 M1 芯片的可运 MacBook Air 上以 fp32 的浮点精度
、18 tok/s 的行已k星速率对于故事妨碍采样。

Karpathy 介绍称，「llama2.c」的灵感来自 llama.cpp，后者由资深开源社区开拓者 Georgi Gerganov 建树
，可能在 MacBook 上运用 4-bit 量化运行第一代 LLaMA 模子 。

对于「llama2.c」 ，它的磨炼代码由 nanoGPT 更正而来，用来磨炼 Llama2 架构的模子。中间是在如下 run.c 中编写 C 推理引擎，不外它当初并非一个破费级库 。下面是部份推理代码。

残缺代码地址 ：https://github.com/karpathy/llama2.c/blob/master/run.c

服从令 Karpathy 颇为惊惶，你可能在（M1）CPU 的纯复线程 C 语言中以 fp32 的交互速率来推理更小（O (~10MB)）的模子。

尽管，他展现自己不试验对于最小规模的 Llama2 模子（70 亿参数）妨碍推理，他估量速率会颇为慢 。

当初 ，Karpathy 在 M1 MacBook Air 上 ，可能以 fp32 的浮点精度
、100tok/s 的速率对于 15M 参数的 288 6 层 6 头的模子妨碍推理。

之后，Karpathy 对于名目妨碍了更新，运用「-O3」妨碍编译可能将 M1 MacBook Air 上的 tok/s 从 18 削减到了 98。这还没完
，运用「-funsafe-math-optimizations」妨碍编译更是将 tok/s 削减到 315。他展现，惟独在 gcc 命令中搜罗更多字符 ，速率就能提升 17.5 倍
。

概况你要问了，这个名目有甚么意思呢？在 Karpathy 看来，在一些较窄的规模（如天生故事）中，人们可能运用极其小的 Transformers 来做幽默的使命。

因此 ，这种可能移植的纯 C 实现概况颇为实用，咱们可能经由重大的措施高交互速率地运行公平巨细的模子（多少万万参数） 。

有网友对于「llama2.c」的开拓历程很感兴趣 ，良多人都市有这样的想法，只是在期待适宜的机缘 ，他们没意见到多少天内就能实现良多使命。

Karpathy 回覆称，自己对于运用 float32 权重块及其上的重大推理代码来天生故事颇为感兴趣。以是他花了全部周六的光阴（从起床不断到睡觉）来写代码 ，而后让名目使命 。

此外，Karpathy 还展现自己将出教学视频。

名目概况

到当初为止，「llama2.c」名目已经在 GitHub 上取患了 1.6k 的 Stars，并在快捷削减。

下面重大介绍一下该名目的运行步骤 。

为了运用纯 C 语言运行一个 baby Llama2 模子，你需要如下的模子魔难点。下载 TinyStories 数据集上磨炼的一个 15M 参数的模子（约莫 58MB），并将它放入默认魔难点目录中 。

wget https://karpathy.ai/llama2c/model.bin -P out

而后编译并运行 C 代码。

gcc -O3 -o run run.c -lm./run out/model.bin

请留意这只是原始 tokens 流。遗憾的是，咱们如今必需经由一个重大的转换封装器来运行 C 代码（惟独 30 行）。

pip install sentencepiecepython run_wrap.py

最后你将看到文本流。在 Karpathy 的 M1 MacBook Air 上 ，运行速率约 100 tok/s ，对于超级原生的 fp32 复线程 C 代码来说还不错。示例输入如下所示 。

更多细节请魔难原名目。

最新评论