[package]
name = "kalosm-learning"
version = "0.4.0"
edition = "2021"
description = "A simplified machine learning library for building off of pretrained models."
license = "MIT/Apache-2.0"
repository = "https://github.com/floneum/floneum"
authors = ["Evan Almloff <evanalmloff@gmail.com>"]
keywords = ["ai", "bert", "nlp", "machine-learning", "transformers"]

[dependencies]
kalosm-language-model.workspace = true
kalosm-learning-macro.workspace = true
candle-core.workspace = true
candle-nn.workspace = true
half = "2.3.1"
rand.workspace = true
kalosm-common = { workspace = true, features = ["candle"] }

[dev-dependencies]
tokio = { version = "1.34.0", features = ["full"] }
rbert.workspace = true
anyhow.workspace = true

[features]
metal = ["candle-core/metal", "candle-nn/metal", "kalosm-common/metal"]