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

kalosm-language-model

A common interface for language models/transformers

8 unstable releases (3 breaking)

0.4.0 Feb 9, 2025
0.3.3 Aug 28, 2024
0.2.1 Feb 28, 2024
0.1.0 Dec 16, 2023

#788 in Machine learning

Download history 81/week @ 2024-10-29 47/week @ 2024-11-05 51/week @ 2024-11-12 65/week @ 2024-11-19 70/week @ 2024-11-26 86/week @ 2024-12-03 140/week @ 2024-12-10 76/week @ 2024-12-17 29/week @ 2024-12-24 105/week @ 2024-12-31 155/week @ 2025-01-07 123/week @ 2025-01-14 85/week @ 2025-01-21 52/week @ 2025-01-28 205/week @ 2025-02-04 85/week @ 2025-02-11

439 downloads per month
Used in 11 crates (8 directly)

MIT/Apache

380KB
8K SLoC

Language Model

This crate provides a unified interface for language models. It supports streaming text, sampling, and embedding.

Usage (with the kalosm-llama implementation 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.complete(prompt);

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

Dependencies

~3–15MB
~174K SLoC