From 52ac194d541f9689084aa9ec70986523e336f35a Mon Sep 17 00:00:00 2001 From: PENG Bo <33809201+BlinkDL@users.noreply.github.com> Date: Tue, 28 Feb 2023 03:27:42 +0800 Subject: [PATCH] Update README.md --- README.md | 2 ++ 1 file changed, 2 insertions(+) diff --git a/README.md b/README.md index 2eb7ae7..d78065a 100644 --- a/README.md +++ b/README.md @@ -8,6 +8,8 @@ So it's combining the best of RNN and transformer - **great performance, fast in **Download RWKV-4 0.1/0.4/1.5/3/7/14B weights**: https://huggingface.co/BlinkDL +**RWKV in 150 lines** (model, inference, text generation): https://github.com/BlinkDL/ChatRWKV/blob/main/RWKV_in_150_lines.py + **ChatRWKV v2:** with "stream" and "split" strategies. **3G VRAM is enough to run RWKV 14B :)** https://github.com/BlinkDL/ChatRWKV/tree/main/v2 ``` os.environ["RWKV_JIT_ON"] = '1'