Cargo.toml

[package]
authors = ["Zhenyuan Zhang <cryscan@umich.edu>"]
categories = ["science", "text-processing"]
description = "An implementation of the RWKV language model in pure WebGPU."
edition = "2021"
exclude = ["assets/", "crates/", "screenshots/"]
homepage = "https://github.com/cryscan/web-rwkv"
keywords = ["deep-learning", "language", "model", "rwkv"]
license = "MIT OR Apache-2.0"
name = "web-rwkv"
repository = "https://github.com/cryscan/web-rwkv"
version = "0.6.36"

# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html

[dependencies]
ahash = "0.8"
anyhow = "1.0"
bitflags = "2.3"
bytemuck = { version = "1.13", features = ["extern_crate_alloc"] }
derive-getters = "0.3"
flume = "0.11.0"
futures = "0.3"
gpp = "0.6.2"
half = { version = "2.2", features = ["bytemuck", "serde"] }
instant = { version = "0.1", features = ["inaccurate", "wasm-bindgen"] }
itertools = "0.12"
lazy_static = "1.4"
log = "0.4"
regex = "1.10"
safetensors = "0.4"
serde = { version = "1.0", features = ["derive"] }
serde_bytes = "0.11.14"
serde_json = "1.0"
thiserror = "1.0"
trait-variant = "0.1"
uid = "0.1"
wasm-bindgen = "0.2"
wgpu = "0.18"

[dependencies.web-rwkv-derive]
path = "crates/web-rwkv-derive"
version = "0.2.4"

[dev-dependencies]
cbor4ii = { version = "0.3.2", features = ["half-f16", "serde1"] }
fastrand = "2.0"
memmap2 = "0.9"
tokio = { version = "1", features = ["full"] }
# wgpu-profiler = "0.14.1"
clap = { version = "4.3", features = ["derive"] }
crossterm = "0.27"
dialoguer = "0.11.0"
ratatui = { version = "0.26", features = ["all-widgets"] }

[profile.release]
lto = false