From 994170685b05eb5aaf8cb4c7484975fd1ca47ecb Mon Sep 17 00:00:00 2001 From: PENG Bo <33809201+BlinkDL@users.noreply.github.com> Date: Sat, 14 Aug 2021 00:06:25 +0800 Subject: [PATCH] Update README.md --- README.md | 13 +++++++++++++ 1 file changed, 13 insertions(+) diff --git a/README.md b/README.md index 7d061f4..792d634 100644 --- a/README.md +++ b/README.md @@ -77,3 +77,16 @@ Red: RWKV ("linear" attention) - VRAM friendly - quite faster when ctx window is Black: MHA_pro (MHA with various tweaks & RWKV-type-FFN) - slow - needs more VRAM - good performance. parameters count: 17.2 vs 18.5 vs 18.5. + +``` +@software{peng_bo_2021_5196578, + author = {PENG Bo}, + title = {BlinkDL/RWKV-LM: 0.01}, + month = aug, + year = 2021, + publisher = {Zenodo}, + version = {0.01}, + doi = {10.5281/zenodo.5196578}, + url = {https://doi.org/10.5281/zenodo.5196578} +} +```