RLlama

RLlama 是量化后的 Llama 7B 语言模型的 Rust 实现。

Llama 7B 是一个非常小但性能优异的语言模型，可以轻松在您的本地机器上运行。

此库使用 Candle 运行 Llama。

使用方法

use kalosm_llama::prelude::*;

#[tokio::main]
async fn main() {
    let mut model = Llama::new().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}");
    }
}