BlinkDL/ChatRWKV

ChatRWKV is like ChatGPT but powered by RWKV (100% RNN) language model, and open source.

What it solves

ChatRWKV provides a way to run a high-quality language model that matches Transformer performance while being significantly faster and using less VRAM. It addresses the efficiency bottlenecks of traditional Transformers by using a Recurrent Neural Network (RNN) architecture.

How it works

It is powered by the RWKV language model, a 100% RNN that scales like a Transformer. The project includes a chat interface and inference scripts that allow users to interact with the models. It supports various inference strategies (such as "stream" and "split") and quantization (INT8) to further reduce memory requirements, enabling large models (like the 14B version) to run on consumer hardware with as little as 3GB of VRAM.

Who it’s for

Developers and AI enthusiasts who want to run powerful LLMs locally with lower hardware requirements or those looking to build their own inference engines based on the RNN architecture.

Highlights

  • RNN Efficiency: Matches Transformer quality but with faster speeds and lower memory usage.
  • Flexible Inference: Supports multiple strategies and INT8 quantization for low VRAM usage.
  • CUDA Acceleration: Includes optional CUDA kernels for significantly faster sequence mode processing.
  • Developer Friendly: Provides simplified scripts (like "RWKV in 150 lines") and API demos to help developers understand and implement the model.

Related

  • Project
  • Dispatch
  • Project
  • Project
  • Project