-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathCargo.toml
More file actions
86 lines (78 loc) · 2.55 KB
/
Copy pathCargo.toml
File metadata and controls
86 lines (78 loc) · 2.55 KB
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
[package]
name = "rustrain"
version = "0.1.0"
edition = "2024"
[dependencies]
# Workspace crates
rustrain-core = { path = "crates/rustrain-core", features = ["tch"] }
rustrain-data = { path = "crates/rustrain-data" }
rustrain-nccl = { path = "crates/rustrain-nccl" }
rustrain-parallel = { path = "crates/rustrain-parallel", features = ["nccl"] }
rustrain-train = { path = "crates/rustrain-train" }
rustrain-toy = { path = "crates/rustrain-toy" }
rustrain-tch-tiny = { path = "crates/rustrain-tch-tiny" }
rustrain-qwen = { path = "crates/rustrain-qwen" }
rustrain-qwen3 = { path = "crates/rustrain-qwen3" }
rustrain-deepseek = { path = "crates/rustrain-deepseek" }
rustrain-deepseek-v4 = { path = "crates/rustrain-deepseek-v4" }
rustrain-glm5 = { path = "crates/rustrain-glm5" }
rustrain-moe = { path = "crates/rustrain-moe" }
rustrain-qwen3-6 = { path = "crates/rustrain-qwen3-6" }
rustrain-server = { path = "crates/rustrain-server" }
rustrain-ipc = { path = "crates/rustrain-ipc" }
# Server deps
axum.workspace = true
tokio.workspace = true
tonic.workspace = true
prost.workspace = true
tower.workspace = true
base64 = "0.22"
futures = "0.3"
tokio-stream = "0.1"
# Ray distributed computing (native Rust SDK) — optional, only needed for Ray mode
rayrust = { version = "0.1.0", optional = true }
# Direct dependencies (used by main.rs, inspect.rs, trainer.rs)
anyhow.workspace = true
clap.workspace = true
ndarray.workspace = true
rand.workspace = true
rand_distr.workspace = true
safetensors.workspace = true
serde.workspace = true
serde_json.workspace = true
tch.workspace = true
toml.workspace = true
tokenizers.workspace = true
tracing.workspace = true
[features]
default = []
ray = ["dep:rayrust"]
[dev-dependencies]
tempfile.workspace = true
[workspace]
members = ["crates/*"]
[workspace.dependencies]
anyhow = "1.0.98"
arrow = { version = "57.0.0", default-features = false, features = ["ipc"] }
axum = "0.8"
chrono = { version = "0.4.41", default-features = false, features = ["clock"] }
clap = { version = "4.5.40", features = ["derive"] }
ndarray = { version = "0.16.1", features = ["serde"] }
prost = "0.13"
rand = "0.8.5"
rand_distr = "0.4.3"
regex = "1.11.3"
safetensors = "0.8.0"
serde = { version = "1.0.219", features = ["derive"] }
serde_json = "1.0.145"
tch = "0.20.0"
tokio = { version = "1", features = ["full"] }
tokenizers = "0.23.1"
toml = "0.8.23"
tonic = "0.12"
tonic-build = "0.12"
tower = { version = "0.5", features = ["util"] }
tracing = "0.1.41"
tracing-appender = "0.2.3"
tracing-subscriber = { version = "0.3.19", features = ["env-filter", "fmt"] }
tempfile = "3.20.0"