[package]
edition = "2024"
rust-version = "1.92"
name = "hermes-llm"
version = "1.7.21"
authors = ["izihawa"]
build = false
autolib = false
autobins = false
autoexamples = false
autotests = false
autobenches = false
description = "LLM training from scratch using Candle"
readme = "README.md"
keywords = [
"llm",
"transformer",
"deep-learning",
"gpt",
"training",
]
categories = ["science"]
license = "MIT"
repository = "https://github.com/SpaceFrontiers/hermes"
resolver = "2"
[features]
accelerate = [
"candle-core/accelerate",
"candle-nn/accelerate",
"candle-transformers/accelerate",
]
cpu = []
cuda = [
"candle-core/cuda",
"candle-nn/cuda",
"candle-transformers/cuda",
]
default = ["cpu"]
flash-attn = [
"cuda",
"candle-flash-attn",
]
metal = [
"candle-core/metal",
"candle-nn/metal",
"candle-transformers/metal",
]
nccl = [
"cuda",
"dep:cudarc",
]
[lib]
name = "hermes_llm"
path = "src/lib.rs"
[[bin]]
name = "hermes-llm"
path = "src/main.rs"
[dependencies.anyhow]
version = "1.0"
[dependencies.candle-core]
version = "0.9"
[dependencies.candle-flash-attn]
version = "0.9"
optional = true
[dependencies.candle-nn]
version = "0.9"
[dependencies.candle-transformers]
version = "0.9"
[dependencies.clap]
version = "4.5"
features = ["derive"]
[dependencies.ctrlc]
version = "~3.4"
[dependencies.cudarc]
version = "0.19"
features = [
"nccl",
"cuda-version-from-build-system",
]
optional = true
[dependencies.flate2]
version = "1.1"
[dependencies.hf-hub]
version = "0.4"
features = [
"ureq",
"rustls-tls",
]
default-features = false
[dependencies.indicatif]
version = "0.18"
[dependencies.pest]
version = "2.7"
[dependencies.pest_derive]
version = "2.7"
[dependencies.rand]
version = "0.9"
[dependencies.rayon]
version = "1.11"
[dependencies.rust-embed]
version = "8"
features = ["include-exclude"]
[dependencies.serde]
version = "1.0"
features = ["derive"]
[dependencies.serde_json]
version = "1.0"
[dependencies.thiserror]
version = "2.0"
[dependencies.tokenizers]
version = "0.22"
features = [
"progressbar",
"fancy-regex",
]
default-features = false
[dependencies.tracing]
version = "0.1"
[dependencies.tracing-subscriber]
version = "0.3"
features = ["env-filter"]
[dependencies.zstd]
version = "0.13"
[dev-dependencies.tempfile]
version = "3.24"