[package]
authors = ["nathanielsimard <nathaniel.simard.42@gmail.com>"]
edition.workspace = true
license.workspace = true
name = "text-classification"
publish = false
version.workspace = true

[lints]
workspace = true

[features]
default = []
f16 = []
flex32 = []
tch-cpu = ["burn/tch"]
tch-gpu = ["burn/tch"]
wgpu = ["burn/wgpu"]
vulkan = ["burn/vulkan"]
flex = ["burn/flex"]
remote = ["burn/remote"]
cuda = ["burn/cuda"]
rocm = ["burn/rocm"]
metal = ["burn/metal"]
ddp = ["burn/distributed"]

[dependencies]
# Burn
burn = { workspace = true, features = [
    "train",
    "optim",
    "tui",
    "sqlite-bundled",
    "metrics",
    "flex",
    "autotune",
    "fusion",
    "std",
] }
log = { workspace = true }

# Tokenizer
tokenizers = { version = "0.22.2", default-features = false, features = [
    "onig",
    "http",
] }

# Utils
derive-new = { workspace = true }
serde = { workspace = true, features = ["std", "derive"] }
