Cargo.toml 4.13 KB
Newer Older
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
# SPDX-License-Identifier: Apache-2.0
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

[package]
Neelay Shah's avatar
Neelay Shah committed
17
name = "dynamo-llm"
18
19
20
21
22
version.workspace = true
edition.workspace = true
authors.workspace = true
license.workspace = true
homepage.workspace = true
23
24
25
repository.workspace = true
readme.workspace = true
description = "Dynamo LLM Library"
26

27
[features]
28
default = []
29
30

cuda_kv = ["dep:cudarc", "dep:ndarray"]
31
llamacpp = ["dep:llama-cpp-2"]
32
mistralrs = ["dep:mistralrs"]
33
python = ["dep:pyo3-async-runtimes", "dep:pythonize"]
34
35
sglang = []
sentencepiece = ["dep:sentencepiece"]
Graham King's avatar
Graham King committed
36
trtllm = []
37
vllm = []
38
39
40
41

cuda = ["mistralrs/cuda", "llama-cpp-2/cuda"]
metal = ["mistralrs/metal", "llama-cpp-2/metal"]
vulkan = ["llama-cpp-2/vulkan"]
42

43
44
[dependencies]
# repo
Neelay Shah's avatar
Neelay Shah committed
45
dynamo-runtime = { workspace = true }
46
47
48
49
50

# workspace
anyhow = { workspace = true }
async-stream = { workspace = true }
async-trait = { workspace = true }
51
async_zmq = { workspace = true }
52
bytes = { workspace = true }
53
chrono = { workspace = true }
54
derive_builder = {workspace = true }
55
either = { workspace = true }
56
futures =  { workspace = true }
57
58
rand = { workspace = true }
prometheus = { workspace = true }
59
serde = { workspace = true }
60
serde_json = { workspace = true }
61
62
63
64
65
66
67
thiserror = { workspace = true }
tokio = { workspace = true }
tokio-stream = { workspace = true }
tokio-util = { workspace = true }
tracing = { workspace = true }
validator = { workspace = true }
uuid = { workspace = true }
68
xxhash-rust = { workspace = true }
69
strum = { workspace = true }
70

Paul Hendricks's avatar
Paul Hendricks committed
71
async-openai = "0.27.2"
72
blake3 = "1"
73
74
bytemuck = "1.22"
derive-getters = "0.5"
75
regex = "1"
76
77
78
79
80
81
82
83
rayon = "1"

# kv_cuda
cudarc = { git = "https://github.com/coreylowman/cudarc.git", rev = "8c52e735b55bf8e979e1a16bd85e3dfe4f87c9fe", features = ["cuda-12040"], optional = true }
ndarray = { version = "0.16", optional = true }
# candle-core = { version = "0.8.3", features = ["cuda"], optional = true }
# half = "2.4.1"

84
85
86
87
88
89
pyo3 = { version = "0.23.3", default-features = false, features = [
  "macros",
  "experimental-async",
  "experimental-inspect",
  "py-clone",
] }
90

91
92
93
94
95
96
# protocols
unicode-segmentation = "1.12"

# http-service
axum = "0.8"

97
98
# mistralrs
indexmap = { version = "2.6" }
99
mistralrs = { git = "https://github.com/EricLBuehler/mistral.rs.git", rev = "aaafc2ef", optional = true }
100

101
102
103
104
# sglang
libc = "0.2"
serde-pickle = "1.2.0"

105
106
107
# llamacpp
llama-cpp-2 = { version = "0.1.86", optional = true }

Biswa Panda's avatar
Biswa Panda committed
108
# tokenizers
109
tokenizers = { version = "0.21.1", default-features = false, features = [
Biswa Panda's avatar
Biswa Panda committed
110
111
  "onig",
  "esaxx_fast",
112
  "rustls-tls",
Biswa Panda's avatar
Biswa Panda committed
113
114
115
116
117
] }
sentencepiece = { version = "0.11.2", optional = true }

# backend
galil-seiferas = { version = "0.1" }
118
toktrie = { version = "0.6.28" }
Biswa Panda's avatar
Biswa Panda committed
119
120
121
122
123
124
125
126
127
128
toktrie_hf_tokenizers =  { version = "0.6.28" }

# preprocessor
bs62 = { version = "0.1" }
erased-serde = { version = "0.4" }
itertools = { version = "0.14.0" }
minijinja = { version = "2.3.1", features = ["loader"] }
minijinja-contrib = { version = "2.3.1", features = ["pycompat"] }
semver = { version = "1", features = ["serde"] }

Graham King's avatar
Graham King committed
129
130
131
# trtllm
serde_repr = "0.1"

132
133
134
135
136
137
138
139
140
141
# python
pyo3-async-runtimes = { version = "0.23.0", optional = true, default-features = false, features = [
  "attributes",
  "testing",
  "tokio-runtime",
  "unstable-streams",
] }
pythonize = { version = "0.23", optional = true }


142
[dev-dependencies]
143
hf-hub = { workspace = true }
144
145
proptest = "1.5.0"
reqwest = { version = "0.12", default-features = false, features = ["json", "stream", "rustls-tls"] }
146
rstest = "0.18.2"
147
rstest_reuse = "0.7.0"
148
tempfile = "3.17.1"
Biswa Panda's avatar
Biswa Panda committed
149
150
151
152
153
154
155
insta = { version = "1.41", features = [
  "glob",
  "json",
  "redactions",
  "filters",
] }

Graham King's avatar
Graham King committed
156
157
158
[build-dependencies]
bindgen = "0.70"
cmake = "0.1"