5 个版本

0.1.1 2023 年 5 月 8 日
0.1.0-rc42023 年 5 月 8 日
0.1.0-rc32023 年 5 月 7 日
0.1.0-rc12023 年 5 月 4 日

#638机器学习

Download history 84/week @ 2024-03-11 61/week @ 2024-03-18 58/week @ 2024-03-25 114/week @ 2024-04-01 163/week @ 2024-04-08 74/week @ 2024-04-15 63/week @ 2024-04-22 66/week @ 2024-04-29 109/week @ 2024-05-06 70/week @ 2024-05-13 79/week @ 2024-05-20 79/week @ 2024-05-27 63/week @ 2024-06-03 69/week @ 2024-06-10 69/week @ 2024-06-17 74/week @ 2024-06-24

每月 283 次下载
用于 llm

MIT/Apache

120KB
2K SLoC

llm

A llama riding a crab, AI-generated

图片由 @darthdeus 提供,使用 Stable Diffusion

Latest version MIT/Apache2 Discord

llm 是一个 Rust 生态系统,用于在大语言模型上运行推理,灵感来自 llama.cpp

主要的 crate 是 llm crate,它封装了 llm-base 和支持模型 crate。

llm 之上,有一个 CLI 应用程序,llm-cli,它提供了一个方便的界面来运行支持模型的推理。此 CLI 可从最新的 GitHub 发布版中获取。

它由 ggml 张量库提供支持,旨在将 Rust 的健壮性和易用性带给大型语言模型的世界。

目前支持以下模型

  • GPT-2
  • GPT-J
  • LLaMA: LLaMA, Alpaca, Vicuna, Koala, GPT4All v1, GPT4-X, Wizard
  • GPT-NeoX: GPT-NeoX, StableLM, Dolly v2 (部分,张量名称不同?)
  • BLOOM: BLOOMZ

入门指南

请确保您已安装 Rust 1.65.0 或更高版本以及 C 工具链[^1]。

llm 是一个 Rust 库,重新导出 llm-base 和模型 crate(例如 bloomgpt2llama)。

llm-cli(二进制名称为 llm)是一个基本应用程序,它提供了对库的 CLI 界面。

注意:为了获得最佳结果,请确保以发布模式构建和运行。调试构建将会非常慢。

使用 cargo 构建

运行

cargo install --git https://github.com/rustformers/llm llm-cli

llm 安装到您的 Cargo bin 目录,该目录 rustup 可能已经添加到您的 PATH 中。

然后可以通过 llm 运行 CLI 应用程序。

从仓库构建

克隆仓库,然后使用以下命令构建:

git clone --recurse-submodules [email protected]:rustformers/llm.git
cargo build --release

生成的二进制文件将位于 target/release/llm[.exe]

也可以直接通过 Cargo 运行,使用

cargo run --release -- <ARGS>

这对于开发很有用。

获取模型

GGML 文件容易获取。有关已测试模型的列表,请参阅已知良好的模型

本项目不支持某些较旧的 GGML 格式,但目标是保持与上游 GGML 项目的功能一致性。有关加载模型的问题,或请求对支持的 GGML 模型类型的支持,请创建一个问题

来自 Hugging Face

Hugging Face 🤗 是开源机器学习的领导者,托管了数百个 GGML 模型。在 Hugging Face 🤗 上搜索 GGML 模型

r/LocalLLaMA

这个 Reddit 社区维护了一个有关 GGML 模型的 维基,包括获取 GGML 模型(大部分来自 Hugging Face 🤗)的链接列表。

LLaMA 原始权重

目前,获取原始权重的唯一合法来源是这个仓库

获取权重后,需要将它们转换为与 ggml 兼容的格式。要实现这一点,请按照以下步骤操作

警告

要运行 Python 脚本,需要 Python 3.9 或 3.10 版本。3.11 在写作时不受支持。

# Convert the model to f16 ggml format
python3 scripts/convert-pth-to-ggml.py /path/to/your/models/7B/ 1

# Quantize the model to 4-bit ggml format
cargo run --release llama quantize /path/to/your/models/7B/ggml-model-f16.bin /path/to/your/models/7B/ggml-model-q4_0.bin q4_0

未来,我们希望提供一种更简便的模型转换方式

注意

llama.cpp 仓库提供了有关如何获取和运行特定模型的信息。

运行

例如,尝试以下提示

llm llama infer -m <path>/ggml-model-q4_0.bin -p "Tell me how cool the Rust programming language is:"

一些其他可以尝试的事情

  • 使用 --help 查看可用选项列表。

  • 如果您有 alpaca-lora 权重,请尝试 repl 模式!

    llm llama repl -m <path>/ggml-alpaca-7b-q4.bin -f examples/alpaca_prompt.txt
    
  • 会话可以加载(--load-session)或保存到文件(--save-session)。要自动加载和保存相同的会话,请使用 --persist-session。这也可以用来缓存提示以减少加载时间。

[^1]: 编译 ggml 需要 C 工具链。通常不需要 C++ 工具链。

Docker

# To build (This will take some time, go grab some coffee):
docker build -t llm .

# To run with prompt:
docker run --rm --name llm -it -v ${PWD}/data:/data -v ${PWD}/examples:/examples llm llama infer -m data/gpt4all-lora-quantized-ggml.bin -p "Tell me how cool the Rust programming language is:"

# To run with prompt file and repl (will wait for user input):
docker run --rm --name llm -it -v ${PWD}/data:/data -v ${PWD}/examples:/examples llm llama repl -m data/gpt4all-lora-quantized-ggml.bin -f examples/alpaca_prompt.txt

Q&A

你为什么要这样做?

这不是我的选择。费里斯出现在我的梦中,并要求我以圣蟹的名义重写它。

说正经的。

快点!我不想引起争吵。你知道是怎么回事,某物某物 内存 某物某物 cargo 很好,别让我说出来,大家都知道这一点。

我坚持。

哎呀!好吧。在看到 llama.cpp 的巨大潜力之后,我首先想到的是将其变成一个库嵌入到我的项目中,看看有多难。我开始深入研究代码,发现重活都是 ggml(一个C库,容易与Rust绑定)做的,整个项目也就大约2k行C++代码(绑定起来并不容易)。在尝试了几次(失败的)将HTTP服务器构建到工具中之后,我意识到如果直接将代码移植到Rust,我会更有生产力,因为我在Rust中更自在。

这是真正的理由吗?

哈哈。当然不是。我只是喜欢收集想象中的互联网积分,形式是小小的星星,人们似乎在我开始无谓的 quest 重写 X 的事情时给我。

这与 llama.cpp 有什么不同?

这是 llama.cpp 的重实现,它与它之外没有任何代码共享,除了 ggml。这样做是出于各种原因

  • llama.cpp 需要C++编译器,这可能在交叉编译到更奇特的平台时造成问题。这样的平台的一个例子是WebAssembly,它可能需要一个非标准的编译器SDK。
  • Rust在开发和开源方面更容易使用;它为编写“大型代码”提供了更好的工具,许多其他作者都可以使用。此外,我们可以轻松地从中受益于更大的Rust生态系统。
  • 我们希望将 ggml 做为一个可选的后端(参见这个问题)。

总的来说,我们希望构建一个与任何其他Rust crate一样容易使用和部署的模型推理解决方案。

哪些应用程序和库使用 llm

应用程序

  • llmcord:使用 llm 生成消息的 Discord 机器人。

依赖关系

~5.5MB
~69K SLoC