[package]
name = "gpt-model"
description = "Pure-Rust inference wrapper for GPT-2 large language models."
version = "0.1.0"
edition = "2021"
license = "AGPL-3.0-only"
readme = "README.md"
repository = "https://gitlab.com/caer/gpt"
authors = [
    "Brandon Sanders",
]

# Don't publish the GPT-2 model export code;
# it's not really part of the crate.
exclude = ["gpt-2-model"]

[dependencies]
# Error handling.
anyhow = "1.0.48"
# N-dimensional array manipulation.
ndarray = "0.15.6"
# Text tokenization; fancy-regex
# required because of negative lookaheads
# in the GPT-2 tokenizer regex.
fancy-regex = "0.10.0"
serde = { version = "1.0", features = ["derive"] }
serde_json = "1.0"
# Model output sampling.
rand = "0.8.5"
# Model execution.
tract-onnx = "0.18.4"