forked from InfiniTensor/InfiniLM
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathCargo.toml
41 lines (34 loc) · 1.02 KB
/
Cargo.toml
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
[workspace]
members = [
"common",
"gguf",
"tensor",
"test-utils",
"models/llama/common",
"models/llama/common-cpu",
"models/llama/opencl",
"models/llama/infini",
"models/llama/nvidia-gpu",
"models/clip/common",
"models/clip/common-cpu",
"models/gpt2/common",
"models/gpt2/common-cpu",
]
resolver = "2"
[workspace.dependencies]
common.path = "common"
gguf.path = "gguf"
tensor.path = "tensor"
causal-lm.path = "causal-lm"
test-utils = { path = "test-utils", default-features = false }
ggus = "0.4"
ndarray-layout = "0.0"
log = "0.4"
regex = "1.11"
itertools = "0.13"
env_logger = "0.11"
build-script-cfg = "0.0"
operators = { git = "https://github.com/YdrMaster/operators-rs", rev = "807ea2b", default-features = false }
search-cl-tools = { git = "https://github.com/InfiniTensor/clrt", rev = "9b6289d" }
search-infini-tools = { git = "https://github.com/InfiniTensor/infini-rt", rev = "f40bcb5" }
search-cuda-tools = { git = "https://github.com/YdrMaster/cuda-driver", rev = "5aec761" }