[package]
name = "llm-tune"
version = "0.1.0"
edition = "2021"
description = "LoRA fine-tuning for Llama models, written from scratch in Rust with burn"
license = "MIT"
repository = "https://git.all.ar/pub/llm-tune"

[dependencies]
burn = { version = "0.21", default-features = false, features = ["std", "ndarray", "autodiff", "store"] }
safetensors = "0.6"
tokenizers = { version = "0.22", default-features = false, features = ["onig"] }
serde = { version = "1", features = ["derive"] }
serde_json = "1"
anyhow = "1"
rand = "0.9"
clap = { version = "4", features = ["derive"] }

[features]
cuda = ["burn/cuda"]
wgpu = ["burn/wgpu"]

[profile.release]
opt-level = 3
lto = "thin"