67 lines
1.4 KiB
Plaintext
67 lines
1.4 KiB
Plaintext
[package]
|
|
name = "rtx-nlg"
|
|
version = "0.1.0"
|
|
edition = "2021"
|
|
description = "Advanced Natural Language Generation for RTX - Production-ready text generation with state-of-the-art algorithms"
|
|
license = "MIT OR Apache-2.0"
|
|
keywords = ["nlp", "nlg", "text-generation", "transformers", "machine-learning"]
|
|
categories = ["science", "text-processing"]
|
|
authors = ["RTX Team"]
|
|
|
|
[dependencies]
|
|
rtx-tensor = { path = "../rtx-tensor" }
|
|
rtx-transformers = { path = "../rtx-transformers" }
|
|
serde = { version = "1.0", features = ["derive"] }
|
|
serde_json = "1.0"
|
|
thiserror = "1.0"
|
|
anyhow = "1.0"
|
|
rayon = "1.8"
|
|
tokio = { version = "1.0", features = ["full"] }
|
|
futures = "0.3"
|
|
dashmap = "5.5"
|
|
indexmap = "2.0"
|
|
regex = "1.10"
|
|
unicode-segmentation = "1.10"
|
|
rand = "0.8"
|
|
rand_distr = "0.4"
|
|
ndarray = "0.15"
|
|
candle-core = "0.3"
|
|
candle-nn = "0.3"
|
|
tch = "0.15"
|
|
tracing = "0.1"
|
|
tracing-subscriber = "0.3"
|
|
metrics = "0.22"
|
|
lru = "0.12"
|
|
num_cpus = "1.16"
|
|
chrono = { version = "0.4", features = ["serde"] }
|
|
async-trait = "0.1"
|
|
tokio-stream = "0.1"
|
|
|
|
[dev-dependencies]
|
|
criterion = { version = "0.5", features = ["html_reports"] }
|
|
proptest = "1.4"
|
|
tempfile = "3.8"
|
|
env_logger = "0.10"
|
|
|
|
[[bench]]
|
|
name = "generation_benchmarks"
|
|
harness = false
|
|
|
|
[[bench]]
|
|
name = "translation_benchmarks"
|
|
harness = false
|
|
|
|
[[bench]]
|
|
name = "summarization_benchmarks"
|
|
harness = false
|
|
|
|
[lib]
|
|
name = "rtx_nlg"
|
|
path = "src/lib.rs"
|
|
|
|
[features]
|
|
default = ["production", "cuda"]
|
|
production = []
|
|
cuda = []
|
|
distributed = []
|
|
streaming = [] |