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.67k stars 217 forks source link

Upgrade Cargo dependencies #434

Closed darxkies closed 1 year ago

darxkies commented 1 year ago

On Arch, rust-bert can not be compiled due to tch errors. With the newer version of tch (0.14.0) it works fine.

OhMyVolk commented 1 year ago

I am also encountering that, already have a PR ready for that https://github.com/guillaume-be/rust-bert/pull/426 Hopefully, it will get merged soon.

guillaume-be commented 1 year ago

As mentioned in https://github.com/guillaume-be/rust-bert/pull/426, unfortunately the latest version of tch causes errors when loading models that have been saved in half precision on a CPU (the CI for this PR is failing). I have raised an issue in the upstream repository (see https://github.com/LaurentMazare/tch-rs/issues/805) but this has not yet been addressed.