Skip to content

Latest commit

 

History

History
29 lines (21 loc) · 696 Bytes

README.md

File metadata and controls

29 lines (21 loc) · 696 Bytes

Text Generation

Note
This example makes use of the HuggingFace datasets library to download the datasets. Make sure you have Python installed on your computer.

The example can be run like so:

CUDA users

git clone https://github.com/tracel-ai/burn.git
cd burn

# Use the --release flag to really speed up training.
export TORCH_CUDA_VERSION=cu121
cargo run --example text-generation --release

Mac users

git clone https://github.com/tracel-ai/burn.git
cd burn

# Use the --release flag to really speed up training.
cargo run --example text-generation --release