candle/candle-examples/examples/bigcode
Laurent Mazare bb3471ea31
Adapt more examples to the updated safetensor api. (#947)
* Simplify the safetensor usage.

* Convert more examples.

* Move more examples.

* Adapt stable-diffusion.
2023-09-23 21:26:03 +01:00
..
README.md Readme gallery (#834) 2023-09-13 09:05:47 +01:00
main.rs Adapt more examples to the updated safetensor api. (#947) 2023-09-23 21:26:03 +01:00

README.md

candle-starcoder: code generation model

StarCoder/BigCode is a LLM model specialized to code generation. The initial model was trained on 80 programming languages.

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)
>     }
> }