Skip to main content

Module bigcode

Module bigcode 

Source
Expand description

BigCode implementation in Rust based on the GPT-BigCode model.

StarCoder/BigCode is a LLM model specialized to code generation. The initial model was trained on 80 programming languages. See “StarCoder: A State-of-the-Art LLM for Code”, Mukherjee et al. 2023

§Running some example

cargo run --example bigcode --release -- --prompt "fn fact(n: u64) -> u64"

> fn fact(n: u64) -> u64  {
>     if n == 0 {
>         1
>     } else {
>         n * fact(n - 1)
>     }
> }

Structs§

Config
GPTBigCode