-
Notifications
You must be signed in to change notification settings - Fork 4
/
Cargo.toml
36 lines (30 loc) · 1.1 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
[package]
name = "rwkvk-rs"
version = "0.1.0"
edition = "2021"
# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
[lib]
path = "src/lib.rs"
[profile.release]
opt-level = 3
lto = "thin"
[dependencies]
anyhow = "1.0"
bytemuck = { version = "1", features = ["extern_crate_alloc"] }
half = { version = "2", features = ["bytemuck", "use-intrinsics"] }
memmap2 = "0"
safetensors = "0.2.8"
clap = { version = "4.1.4", features = ["derive"], optional = true }
tokenizers = { version = "0.13", default-features = false, features = ["esaxx_fast", "fancy-regex", "onig"], optional = true }
cblas = { version = "0.4", optional = true }
blas-src = { version = "0.8", features = ["openblas"], optional = true }
# blis-src = { version = "0.2", features = ["static", "pthreads"], optional = true }
openblas-src = { version = "0.10", optional = true }
[features]
build-binary = ["clap", "tokenizers"]
# rust-blas = ["cblas", "blas-src", "blis-src"]
rust-blas = ["cblas", "blas-src", "openblas-src"]
[[bin]]
name = "rwkvk-rs"
path = "src/main.rs"
required-features = ["build-binary"]