[package]
edition = "2021"
name = "combs-models"
version = "0.2.2"
build = false
autolib = false
autobins = false
autoexamples = false
autotests = false
autobenches = false
description = "Combs Engine model architecture registry (Llama family)"
readme = false
license = "MIT OR Apache-2.0"
[lib]
name = "combs_models"
path = "src/lib.rs"
[[test]]
name = "attn_cliff"
path = "tests/attn_cliff.rs"
[[test]]
name = "gemma_parity"
path = "tests/gemma_parity.rs"
[[test]]
name = "kv_cache"
path = "tests/kv_cache.rs"
[[test]]
name = "phi_fused"
path = "tests/phi_fused.rs"
[[test]]
name = "prefill_cpu"
path = "tests/prefill_cpu.rs"
[[test]]
name = "qwen_bias"
path = "tests/qwen_bias.rs"
[dependencies.burn]
version = "0.21.0"
features = [
"wgpu",
"fusion",
"std",
]
default-features = false
[dependencies.burn-cubecl]
version = "0.21.0"
features = ["fusion"]
[dependencies.burn-cubecl-fusion]
version = "0.21.0"
[dependencies.burn-fusion]
version = "0.21.0"
[dependencies.burn-ir]
version = "0.21.0"
[dependencies.combs-core]
version = "0.2.2"
[dependencies.combs-formats]
version = "0.2.2"
[dependencies.cubecl]
version = "0.10.0"
features = [
"wgpu",
"std",
]
default-features = false
[dependencies.thiserror]
version = "2"
[dev-dependencies.anyhow]
version = "1"
[dev-dependencies.burn]
version = "0.21.0"
features = [
"ndarray",
"std",
]
default-features = false
[dev-dependencies.tokenizers]
version = "0.22"
features = ["fancy-regex"]
default-features = false