guillaume-be / rust-bert

Rust native ready-to-use NLP pipelines and transformer-based models (BERT, DistilBERT, GPT2,...)
https://docs.rs/crate/rust-bert
Apache License 2.0
2.6k stars 215 forks source link

Change `generate` return type to Result #437

Closed guillaume-be closed 10 months ago

guillaume-be commented 10 months ago

Text generation traits and pipelines (including conversation, summarization and translation) now return a Result for improved error handling

Fixes https://github.com/guillaume-be/rust-bert/issues/436