Text Generation
PyTorch
12 languages
causal-lm
rwkv
rwkv-5-world / README.md
BlinkDL's picture
Update README.md
1b3f6b8
---
language:
- en
- zh
- de
- fr
- es
- pt
- ru
- it
- ja
- ko
- vi
- ar
tags:
- pytorch
- text-generation
- causal-lm
- rwkv
license: apache-2.0
datasets:
- EleutherAI/pile
- togethercomputer/RedPajama-Data-1T
---
# RWKV-5 World (Training in Progress)
Note https://huggingface.co/BlinkDL/rwkv-4-world are the best models as of now, because I am still training RWKV-5.
Use rwkv pip package 0.8.7+ for RWKV-5 inference.
## Model Description
RWKV-5 trained on 100+ world languages (70% English, 15% multilang, 15% code).
World = Some_Pile + Some_RedPajama + Some_OSCAR + All_Wikipedia + All_ChatGPT_Data_I_can_find
RWKV-5 training: set --my_testing "r" in latest RWKV-LM
RWKV-5 inference implementation reference: https://github.com/BlinkDL/ChatRWKV/blob/main/RWKV_v5_demo.py
0.1B world:
RWKV-5 37% trained = LAMBADA ppl 18.1 acc 42.93%
RWKV-4 100% trained = LAMBADA ppl 25.5 acc 36.29%