huggingface / nanotron

Minimalistic large language model 3D-parallelism training
Apache License 2.0
1.23k stars 122 forks source link

Make custom models a bit easier to use #53

Closed thomwolf closed 9 months ago

thomwolf commented 9 months ago

This way they can be used in run_generate as well