# THIS FILE IS AUTOMATICALLY GENERATED BY CARGO
#
# When uploading crates to the registry Cargo will automatically
# "normalize" Cargo.toml files for maximal compatibility
# with all versions of Cargo and also rewrite `path` dependencies
# to registry (e.g., crates.io) dependencies.
#
# If you are reading this file be aware that the original Cargo.toml
# will likely look very different (and much more reasonable).
# See Cargo.toml.orig for the original contents.

[package]
edition = "2021"
name = "finetype-train"
version = "0.6.42"
build = false
autolib = false
autobins = false
autoexamples = false
autotests = false
autobenches = false
description = "Training infrastructure for FineType — Sense, Entity, and Model2Vec training via Candle"
homepage = "https://meridian.online/projects/finetype/"
readme = "README.md"
license = "MIT"
repository = "https://github.com/meridian-online/finetype"

[features]
cuda = [
    "candle-core/cuda",
    "candle-nn/cuda",
]
default = ["tui"]
metal = [
    "candle-core/metal",
    "candle-nn/metal",
]
tui = [
    "dep:ratatui",
    "dep:crossterm",
]

[lib]
name = "finetype_train"
path = "src/lib.rs"

[[bin]]
name = "predict_multibranch"
path = "src/bin/predict_multibranch.rs"

[[bin]]
name = "prepare-model2vec"
path = "src/bin/prepare_model2vec.rs"

[[bin]]
name = "prepare-sense-data"
path = "src/bin/prepare_sense_data.rs"

[[bin]]
name = "train-entity-classifier"
path = "src/bin/train_entity.rs"

[[bin]]
name = "train-sense-model"
path = "src/bin/train_sense.rs"

[[bin]]
name = "train-sibling-context"
path = "src/bin/train_sibling_context.rs"

[[bin]]
name = "tui-demo"
path = "src/bin/tui_demo.rs"

[dependencies.anyhow]
version = "1.0"

[dependencies.candle-core]
version = "0.8"

[dependencies.candle-nn]
version = "0.8"

[dependencies.clap]
version = "4.5"
features = ["derive"]

[dependencies.crossterm]
version = "0.28"
optional = true

[dependencies.csv]
version = "1.3"

[dependencies.duckdb]
version = "=1.10503.1"
features = ["bundled"]
default-features = true

[dependencies.finetype-core]
version = "0.6.0"

[dependencies.finetype-model]
version = "0.6.0"

[dependencies.rand]
version = "0.8"

[dependencies.ratatui]
version = "0.29"
optional = true

[dependencies.safetensors]
version = "0.4"

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

[dependencies.serde_json]
version = "1.0"

[dependencies.serde_yaml]
version = "0.9"

[dependencies.tracing]
version = "0.1"

[dependencies.tracing-subscriber]
version = "0.3"
features = ["env-filter"]

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