mirror of
https://github.com/guillaume-be/rust-bert.git
synced 2024-09-11 12:55:34 +03:00
78 lines
1.6 KiB
TOML
78 lines
1.6 KiB
TOML
[package]
|
|
name = "rust-bert"
|
|
version = "0.16.0"
|
|
authors = ["Guillaume Becquin <guillaume.becquin@gmail.com>"]
|
|
edition = "2018"
|
|
description = "Ready-to-use NLP pipelines and transformer-based models (BERT, DistilBERT, GPT2,...)"
|
|
repository = "https://github.com/guillaume-be/rust-bert"
|
|
documentation = "https://docs.rs/rust-bert"
|
|
license = "Apache-2.0"
|
|
readme = "README.md"
|
|
keywords = ["nlp", "deep-learning", "machine-learning", "transformers", "translation"]
|
|
|
|
# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
|
|
|
|
[lib]
|
|
name = "rust_bert"
|
|
path = "src/lib.rs"
|
|
crate-type = ["lib"]
|
|
|
|
[[bin]]
|
|
name = "convert-tensor"
|
|
path = "src/convert-tensor.rs"
|
|
doc = false
|
|
|
|
[[bench]]
|
|
name = "sst2_benchmark"
|
|
harness = false
|
|
|
|
[[bench]]
|
|
name = "squad_benchmark"
|
|
harness = false
|
|
|
|
[[bench]]
|
|
name = "summarization_benchmark"
|
|
harness = false
|
|
|
|
[[bench]]
|
|
name = "translation_benchmark"
|
|
harness = false
|
|
|
|
[[bench]]
|
|
name = "generation_benchmark"
|
|
harness = false
|
|
|
|
[[bench]]
|
|
name = "tensor_operations_benchmark"
|
|
harness = false
|
|
|
|
[profile.bench]
|
|
opt-level = 3
|
|
|
|
[features]
|
|
doc-only = ["tch/doc-only"]
|
|
all-tests = []
|
|
|
|
[package.metadata.docs.rs]
|
|
features = ["doc-only"]
|
|
|
|
[dependencies]
|
|
rust_tokenizers = "~7.0.0"
|
|
tch = "~0.6.1"
|
|
serde_json = "1.0.68"
|
|
serde = { version = "1.0.130", features = ["derive"] }
|
|
dirs = "4.0.0"
|
|
ordered-float = "2.8.0"
|
|
cached-path = "0.5.1"
|
|
lazy_static = "1.4.0"
|
|
uuid = { version = "0.8.2", features = ["v4"] }
|
|
thiserror = "1.0.30"
|
|
half = "1.7.1"
|
|
|
|
[dev-dependencies]
|
|
anyhow = "1.0.44"
|
|
csv = "1.1.6"
|
|
criterion = "0.3.5"
|
|
torch-sys = "~0.6.1"
|
|
tempfile = "3.2.0"
|