blob: 61e0a963c944fdb28cc45eac1d9a8d247afd8571 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
|
[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"
|
