mirror of
https://github.com/hanzoai/musetalk-bench.git
synced 2026-11-03 01:49:22 +00:00
135 lines
4.5 KiB
TOML
135 lines
4.5 KiB
TOML
[workspace]
|
|
# Only the hanzo-* crates are workspace members. The legacy hanzo-ml-* crates are an orphaned
|
|
# duplicate cluster (no hanzo crate depends on them); keeping them out of the build drops the
|
|
# `hanzo-ml` name from the workspace and resolves the hanzo-ml-pyo3 vs hanzo-ml-pyo3 `links="python"`
|
|
# conflict. The dirs remain on disk for now and can be deleted in a follow-up.
|
|
members = [
|
|
"hanzo-datasets",
|
|
"hanzo-ml",
|
|
"hanzo-ml-examples",
|
|
"hanzo-ml-pyo3",
|
|
"hanzo-ml-wasm-examples/*",
|
|
"hanzo-ml-wasm-tests",
|
|
"hanzo-nn",
|
|
"hanzo-training",
|
|
"hanzo-transformers",
|
|
"tensor-tools",
|
|
]
|
|
exclude = [
|
|
"hanzo-ml-book",
|
|
"hanzo-flash-attn",
|
|
"hanzo-flash-attn-v3",
|
|
"hanzo-kernels",
|
|
"hanzo-metal-kernels",
|
|
"hanzo-rocm-kernels",
|
|
"hanzo-onnx",
|
|
"hanzo-flash-attn",
|
|
"hanzo-kernels",
|
|
"hanzo-metal-kernels",
|
|
"hanzo-ml-book",
|
|
"hanzo-onnx",
|
|
"hanzo-ug",
|
|
]
|
|
resolver = "2"
|
|
|
|
[workspace.package]
|
|
version = "0.9.2"
|
|
edition = "2021"
|
|
description = "Minimalist ML framework."
|
|
repository = "https://github.com/hanzoai/ml"
|
|
keywords = ["blas", "tensor", "machine-learning"]
|
|
categories = ["science"]
|
|
license = "MIT OR Apache-2.0"
|
|
|
|
[workspace.dependencies]
|
|
rocm-rs = { version = "0.5", default-features = false, features = ["macros"] }
|
|
hanzo-rocm-kernels = { path = "./hanzo-rocm-kernels", version = "0.10.2" }
|
|
ab_glyph = "0.2.23"
|
|
accelerate-src = { version = "0.3.2" }
|
|
anyhow = { version = "1", features = ["backtrace"] }
|
|
byteorder = "1.4.3"
|
|
chrono = "0.4"
|
|
# Hanzo-renamed workspace dependencies
|
|
hanzo-datasets = { path = "./hanzo-datasets", version = "0.9.2" }
|
|
hanzo-flash-attn = { path = "./hanzo-flash-attn", version = "0.9.2" }
|
|
hanzo-kernels = { path = "./hanzo-kernels", version = "0.9.2" }
|
|
hanzo-metal-kernels = { path = "./hanzo-metal-kernels", version = "0.9.2" }
|
|
hanzo-ml = { path = "./hanzo-ml", version = "0.9.2" }
|
|
hanzo-nn = { path = "./hanzo-nn", version = "0.9.2" }
|
|
hanzo-onnx = { path = "./hanzo-onnx", version = "0.9.2" }
|
|
hanzo-transformers = { path = "./hanzo-transformers", version = "0.9.2" }
|
|
hanzo-ug = { path = "./hanzo-ug", version = "0.9.2" }
|
|
clap = { version = "4.2.4", features = ["derive"] }
|
|
criterion = { version = "0.8", default-features = false }
|
|
cudarc = { version = "0.19.1", features = [
|
|
"std",
|
|
"cublas",
|
|
"cublaslt",
|
|
"curand",
|
|
"driver",
|
|
"nvrtc",
|
|
"f16",
|
|
"f8",
|
|
"cuda-version-from-build-system",
|
|
"dynamic-linking",
|
|
], default-features = false }
|
|
fancy-regex = "0.17.0"
|
|
gemm = { version = "0.19.0", features = ["wasm-simd128-enable"] }
|
|
hf-hub = "0.4.1"
|
|
half = { version = "2.5.0", features = [
|
|
"num-traits",
|
|
"use-intrinsics",
|
|
"rand_distr",
|
|
] }
|
|
float8 = { version = "0.7.0", features = ["num-traits", "rand_distr"] }
|
|
hound = "3.5.1"
|
|
image = { version = "0.25.2", default-features = false, features = [
|
|
"jpeg",
|
|
"png",
|
|
] }
|
|
imageproc = { version = "0.26.0", features = [
|
|
"text",
|
|
], default-features = false }
|
|
intel-mkl-src = { version = "0.8.1", features = ["mkl-static-lp64-iomp"] }
|
|
libc = { version = "0.2.147" }
|
|
libm = { version = "0.2.15" }
|
|
log = "0.4"
|
|
memmap2 = { version = "0.9.3", features = ["stable_deref_trait"] }
|
|
num_cpus = "1.15.0"
|
|
num-traits = "0.2.15"
|
|
parquet = "57"
|
|
rand = "0.9.0"
|
|
rand_distr = "0.5.1"
|
|
rayon = "1.7.0"
|
|
safetensors = "0.7.0"
|
|
serde = { version = "1.0.171", features = ["derive"] }
|
|
serde_plain = "1.0.2"
|
|
serde_json = "1.0.99"
|
|
thiserror = "2"
|
|
tokenizers = { version = "0.22.0", default-features = false }
|
|
tracing = "0.1.37"
|
|
tracing-chrome = "0.7.1"
|
|
tracing-subscriber = "0.3.7"
|
|
ug = "0.5.0"
|
|
ug-cuda = "0.5.0"
|
|
ug-metal = "0.5.0"
|
|
yoke = { version = "0.8.1", features = ["derive"] }
|
|
zip = { version = "7.2.0", default-features = false }
|
|
objc2-metal = { version = "0.3.1" }
|
|
objc2-foundation = { version = "0.3.1" }
|
|
objc2 = { version = "0.6" }
|
|
objc2-metal-performance-shaders = { version = "0.3.2", default-features = false, features = ["std", "MPSCore", "MPSCoreTypes", "MPSNDArray"] }
|
|
objc2-metal-performance-shaders-graph = { version = "0.3.2", default-features = false, features = ["std", "objc2-metal-performance-shaders", "MPSGraph", "MPSGraphCore", "MPSGraphConvolutionOps", "MPSGraphMemoryOps", "MPSGraphTensor", "MPSGraphTensorData", "MPSGraphOperation"] }
|
|
|
|
[profile.release-with-debug]
|
|
inherits = "release"
|
|
debug = true
|
|
|
|
[workspace.lints.clippy]
|
|
# Stylistic lints in inherited model/bench code; allowed so CI clippy -D warnings is green.
|
|
explicit_counter_loop = "allow"
|
|
unnecessary_unwrap = "allow"
|
|
manual_checked_ops = "allow"
|
|
needless_borrow = "allow"
|
|
large_const_arrays = "allow"
|