#language-model #nlp #llm #ai #llama #mistral

kalosm-language-model

语言模型/转换器的通用接口

6 个版本

新版本 0.3.2 2024 年 8 月 14 日
0.3.1 2024 年 8 月 13 日
0.2.1 2024 年 2 月 28 日
0.1.0 2023 年 12 月 16 日

#762机器学习

Download history • Rust 包仓库 4/week @ 2024-04-26 • Rust 包仓库 16/week @ 2024-05-17 • Rust 包仓库 25/week @ 2024-05-24 • Rust 包仓库 23/week @ 2024-05-31 • Rust 包仓库 19/week @ 2024-06-07 • Rust 包仓库 26/week @ 2024-06-14 • Rust 包仓库 42/week @ 2024-06-21 • Rust 包仓库 19/week @ 2024-06-28 • Rust 包仓库 10/week @ 2024-07-05 • Rust 包仓库 27/week @ 2024-07-12 • Rust 包仓库 17/week @ 2024-07-19 • Rust 包仓库 19/week @ 2024-07-26 • Rust 包仓库 21/week @ 2024-08-02 • Rust 包仓库 355/week @ 2024-08-09 • Rust 包仓库

每月 419 次下载
用于 11 个 crate (8 个直接使用)

MIT/Apache

295KB
7K SLoC

语言模型

本 crate 提供语言模型的一个统一接口。它支持流式文本、采样和嵌入。

使用示例(与 RPhi 实现crate一起使用)

use kalosm::language::*;

#[tokio::main]
async fn main() {
    let mut model = Llama::phi_3().await.unwrap();
    let prompt = "The capital of France is ";
    let mut result = model.stream_text(prompt).await.unwrap();

    print!("{prompt}");
    while let Some(token) = result.next().await {
        print!("{token}");
    }
}

依赖项

~31–49MB
~796K SLoC