rust-bert 0.21.0

Ready-to-use NLP pipelines and language models
Documentation
[[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"

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

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

[dependencies.dirs]
optional = true
version = "4"

[dependencies.half]
version = "2"

[dependencies.lazy_static]
optional = true
version = "1"

[dependencies.ndarray]
optional = true
version = "0.15"

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

[dependencies.ort]
default-features = false
features = ["half"]
optional = true
version = "1.14.8"

[dependencies.regex]
version = "1.6"

[dependencies.rust_tokenizers]
version = "8.1"

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

[dependencies.serde_json]
version = "1"

[dependencies.tch]
version = "0.13.0"

[dependencies.thiserror]
version = "1"

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

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

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

[dev-dependencies.itertools]
version = "0.10"

[dev-dependencies.ort]
features = ["load-dynamic"]
version = "1.14.8"

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

[dev-dependencies.tokio]
features = ["sync", "rt-multi-thread", "macros"]
version = "1.24"

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

[dev-dependencies.tracing-subscriber]
default-features = false
features = ["env-filter", "fmt"]
version = "0.3"

[features]
all-tests = []
default = ["remote"]
doc-only = ["tch/doc-only"]
download-libtorch = ["tch/download-libtorch"]
onnx = ["ort", "ndarray"]
remote = ["cached-path", "dirs", "lazy_static"]

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

[package]
authors = ["Guillaume Becquin <guillaume.becquin@gmail.com>"]
build = "build.rs"
description = "Ready-to-use NLP pipelines and language models"
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.21.0"
[package.metadata.docs.rs]
features = ["doc-only"]
[profile.bench]
opt-level = 3