ace-step-rs/Cargo.toml at master · Marenz/ace-step-rs · GitHub

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
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
[package]
name = "ace-step-rs"
version = "0.2.0"
edition = "2024"
description = "ACE-Step v1.5 music generation in pure Rust via candle"
license = "MIT"
repository = "https://github.com/marenz/ace-step-rs"

[dependencies]
# ML framework (local candle fork with matmul+col2im ConvTranspose1d + optional cuDNN)
candle-core = { path = "/home/marenz/repos/candle/candle-core" }
candle-nn = { path = "/home/marenz/repos/candle/candle-nn" }
candle-transformers = { path = "/home/marenz/repos/candle/candle-transformers" }

# Tokenization (Qwen3 tokenizer)
tokenizers = "0.21"

# Audio I/O
hound = "3.5"

# Serialization
serde = { version = "1", features = ["derive"] }
serde_json = "1"

# Error handling
thiserror = "2"

# Logging
tracing = "0.1"

# Random number generation
rand = "0.9"

# Model downloads
hf-hub = "0.4"

# Float16
half = "2"

# CUDA runtime (VRAM queries) — only used when the cuda feature is enabled
cudarc = { version = "0.19.1", features = ["runtime", "cuda-version-from-build-system", "dynamic-linking"], optional = true }

# Numerical traits
num-traits = "0.2"

# Zip archive reading (for loading .pt files)
zip = "2"

# Async runtime (GenerationManager uses tokio channels + spawn_blocking)
tokio = { version = "1", features = ["rt", "rt-multi-thread", "sync", "macros", "net", "io-util"] }

# XDG base directories (for spool path)
dirs = "5"

# CLI argument parsing (binary only)
clap = { version = "4", features = ["derive"] }
anyhow = "1"

# Logging output
tracing-subscriber = { version = "0.3", features = ["env-filter"] }

# CLI audio playback + terminal input (behind "cli" feature)
rodio = { version = "0.20", optional = true }
cpal = { version = "0.15", optional = true }
ringbuf = { version = "0.4", optional = true }
crossterm = { version = "0.28", optional = true }
ctrlc = { version = "3", optional = true }

[dev-dependencies]
approx = "0.5"
tempfile = "3"

[features]
default = ["cuda"]
cuda = ["candle-core/cuda", "candle-nn/cuda", "candle-transformers/cuda", "dep:cudarc"]
cudnn = ["cuda", "candle-core/cudnn"]
metal = ["candle-core/metal", "candle-nn/metal", "candle-transformers/metal"]
cli = ["dep:rodio", "dep:cpal", "dep:ringbuf", "dep:crossterm", "dep:ctrlc"]

# Optional audio encoding
audio-ogg = ["dep:vorbis_rs"]
audio-mp3 = ["dep:mp3lame-encoder"]
audio-all = ["audio-ogg", "audio-mp3"]

[target.'cfg(not(all(target_os = "windows", target_arch = "aarch64")))'.dependencies]
vorbis_rs = { version = "0.5", optional = true }

[dependencies.mp3lame-encoder]
version = "0.2"
optional = true

[[example]]
name = "radio_daemon"
required-features = ["cli"]

[[bin]]
name = "generation-daemon"
path = "src/bin/generation-daemon.rs"
required-features = ["audio-ogg"]

[[bin]]
name = "ace-step-client"
path = "src/bin/ace-step-client.rs"