forked from oppiliappan/rt-bench
-
Notifications
You must be signed in to change notification settings - Fork 2
/
Cargo.toml
36 lines (30 loc) · 912 Bytes
/
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
[package]
name = "benches"
version = "0.1.0"
edition = "2021"
# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
[dependencies]
ort = { git = "https://github.com/bloopai/ort", branch = "env-builder-telemetry", features = ["load-dynamic"] }
tokenizers = {version="=0.13.3", default-features=false, features=["onig"]}
ndarray = "0.15"
anyhow = "1.0.72"
llm = { git = "https://github.com/bloopai/llm", branch = "batch-metal-kernels", default-features = false, features = ["bert"], optional = true }
hf-hub = "0.2.0"
candle-core = "0.1.2"
candle-nn = "0.1.2"
tracing = "0.1.37"
serde = "1.0.186"
serde_json = "1.0.105"
libc = { version = "0.2.147" }
accelerate-src = { version = "0.3.2" }
[dev-dependencies]
criterion = "0.3"
[[bench]]
name = "benchmarko"
harness = false
[[bin]]
name = "sim"
path = "src/sim.rs"
[[bin]]
name = "print"
path = "src/print_embed.rs"