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

Update tch to 0.15 #443

Closed blmarket closed 7 months ago

blmarket commented 8 months ago

It allows to run in system with libtorch v2.2.0, upgrading from v2.1.0

Also it seems torch-sys dev dependency is unnecessary, so just removed.

guillaume-be commented 7 months ago

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