Web-RWKV
This is an implementation of the language model of RWKV in pure WebGPU.
Compile and Run
- Install Rust.
- Run
cargo r -r --example generationto generate 100 tokens and test the time cost.
Credits
- Tokenizer is implemented by @koute.
This is an implementation of the language model of RWKV in pure WebGPU.
cargo r -r --example generation to generate 100 tokens and test the time cost.