Updated to tch 0.3.0
[package]
name = "rust-bert"
-version = "0.11.0"
+version = "0.12.0"
authors = ["Guillaume Becquin <guillaume.becquin@gmail.com>"]
edition = "2018"
description = "Ready-to-use NLP pipelines and transformer-based models (BERT, DistilBERT, GPT2,...)"
@@ -54,7 +54,7 @@ features = ["doc-only"]
[dependencies]
rust_tokenizers = "~6.0.0"
-tch = "~0.2.1"
+tch = "~0.3.0"
serde_json = "1.0.59"
serde = { version = "1.0.117", features = ["derive"] }
dirs = "3.0.1"
@@ -69,4 +69,4 @@ thiserror = "1.0.21"
anyhow = "1.0.33"
csv = "1.1.3"
criterion = "0.3.3"
-torch-sys = "0.2.1"
+torch-sys = "0.3.0"
