`tch 0.14.0` update - Githubissues

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.51k stars 211 forks source link

Closed guillaume-be closed 7 months ago

guillaume-be commented 7 months ago

Added optional kind parameter to specify the model weight precision to most pipeline configurations. If not provided, will default to full precision on CPU, or the serialized weights precision otherwise.
Fixed issue with GPT-J that was incorrectly tracking the gradients for the attention bias
Upgraded to torch 2.1 (via tch 0.14.0).