forked from rustformers/llm
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Cargo.toml
52 lines (41 loc) · 1.37 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
42
43
44
45
46
47
48
49
50
51
52
[package]
edition = "2021"
name = "llm-cli"
version = "0.2.0-dev"
repository = { workspace = true }
license = { workspace = true }
description = "A CLI for running inference on supported Large Language Models. Powered by the `llm` library."
readme = "../../README.md"
[[bin]]
name = "llm"
path = "src/main.rs"
[dependencies]
llm = { path = "../../crates/llm", version = "0.2.0-dev", default-features = false, features = ["models"] }
bytesize = { workspace = true }
env_logger = { workspace = true }
log = { workspace = true }
rand = { workspace = true }
rustyline = { workspace = true }
spinoff = { workspace = true }
clap = { workspace = true }
bincode = "1.3.3"
num_cpus = "1.15.0"
color-eyre = { version = "0.6.2", default-features = false }
zstd = { version = "0.12", default-features = false }
tracing-subscriber = {workspace = true }
tracing = { workspace = true}
tracing-appender = "0.2.2"
# TEMPORARY: This was introduced in Rust 1.70, but our MSRV is below this.
# Remove this once we bump our MSRV to 1.70.
is-terminal = "0.4"
llm-samplers = { workspace = true }
[dev-dependencies]
rusty-hook = "^0.11.2"
[features]
default = ["tokenizers-remote"]
tokenizers-remote = ["llm/tokenizers-remote"]
cublas = ["llm/cublas"]
clblast = ["llm/clblast"]
metal = ["llm/metal"]
# Falcon is off by default. See `llm_falcon`'s module documentation for more information.
falcon = ["llm/falcon"]