kalosm-language-model 0.2.1

A common interface for language models/transformers
Documentation

Language Model

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

Usage (with the RPhi implementation crate)

use rphi::prelude::*;

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