Skip to content

Commit

Permalink
Update docs.
Browse files Browse the repository at this point in the history
  • Loading branch information
cryscan committed May 2, 2024
1 parent b336575 commit e8507b8
Show file tree
Hide file tree
Showing 2 changed files with 9 additions and 2 deletions.
7 changes: 5 additions & 2 deletions Cargo.toml
Original file line number Diff line number Diff line change
Expand Up @@ -11,13 +11,12 @@ name = "web-rwkv"
repository = "https://github.com/cryscan/web-rwkv"
version = "0.8.0"

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

[dependencies]
ahash = "0.8"
anyhow = "1.0"
bytemuck = { version = "1.13", features = ["extern_crate_alloc"] }
derive-getters = "0.3"
document-features = "0.2.8"
flume = { version = "0.11.0" }
futures = "0.3"
gpp = "0.6.2"
Expand Down Expand Up @@ -61,9 +60,13 @@ simple_logger = { version = "5.0.0", features = ["stderr"] }
[features]
default = ["runtime", "subgroup-ops", "tokio-multi-thread"]

## Enables async GPU command building. Can be faster on some Xeon cpus but slower on others.
async-build = []
## Enables `runtime` API, which essentially doubles the inference speed comparing to the old API.
runtime = []
## Enables subgroup operations in the kernels. Accelerates the inference on some device.
subgroup-ops = []
## Enables tokio's multi-threaded runtime. Doesn't work on web platforms.
tokio-multi-thread = ["tokio/rt-multi-thread"]

[[example]]
Expand Down
4 changes: 4 additions & 0 deletions src/lib.rs
Original file line number Diff line number Diff line change
Expand Up @@ -29,6 +29,10 @@
//! - State caching or management system.
//! - Python (or any other languages) binding.
//! - Runtime. Without a runtime makes it easy to be integrated into any applications from servers, front-end apps (yes, `web-rwkv` can run in browser) to game engines.
//!
//! ## Crate Features
//!
#![doc = document_features::document_features!()]

pub mod context;
pub mod model;
Expand Down

0 comments on commit e8507b8

Please sign in to comment.