rust-bert 0.12.0

[[bench]]
harness = false
name = "sst2_benchmark"

[[bench]]
harness = false
name = "squad_benchmark"

[[bench]]
harness = false
name = "summarization_benchmark"

[[bench]]
harness = false
name = "translation_benchmark"

[[bench]]
harness = false
name = "generation_benchmark"

[[bench]]
harness = false
name = "tensor_operations_benchmark"

[[bin]]
doc = false
name = "convert-tensor"
path = "src/convert-tensor.rs"
[dependencies.cached-path]
version = "0.4.5"

[dependencies.dirs]
version = "3.0.1"

[dependencies.itertools]
version = "0.9.0"

[dependencies.lazy_static]
version = "1.4.0"

[dependencies.ordered-float]
version = "2.0.0"

[dependencies.rust_tokenizers]
version = "~6.1.0"

[dependencies.serde]
features = ["derive"]
version = "1.0.117"

[dependencies.serde_json]
version = "1.0.59"

[dependencies.tch]
version = "~0.3.0"

[dependencies.thiserror]
version = "1.0.22"

[dependencies.uuid]
features = ["v4"]
version = "0.8.1"
[dev-dependencies.anyhow]
version = "1.0.34"

[dev-dependencies.criterion]
version = "0.3.3"

[dev-dependencies.csv]
version = "1.1.5"

[dev-dependencies.tempfile]
version = "3.1.0"

[dev-dependencies.torch-sys]
version = "0.3.0"

[features]
all-tests = []
doc-only = ["tch/doc-only"]

[lib]
crate-type = ["lib"]
name = "rust_bert"
path = "src/lib.rs"

[package]
authors = ["Guillaume Becquin <guillaume.becquin@gmail.com>"]
description = "Ready-to-use NLP pipelines and transformer-based models (BERT, DistilBERT, GPT2,...)"
documentation = "https://docs.rs/rust-bert"
edition = "2018"
keywords = ["nlp", "deep-learning", "machine-learning", "transformers", "translation"]
license = "Apache-2.0"
name = "rust-bert"
readme = "README.md"
repository = "https://github.com/guillaume-be/rust-bert"
version = "0.12.0"
[package.metadata.docs.rs]
features = ["doc-only"]
[profile.bench]
opt-level = 3