|
--- |
|
language: |
|
- en |
|
- zh |
|
- fr |
|
- es |
|
- de |
|
- pt |
|
- ru |
|
- it |
|
- ja |
|
- ko |
|
- vi |
|
- ar |
|
tags: |
|
- pytorch |
|
- text-generation |
|
- causal-lm |
|
- rwkv |
|
license: apache-2.0 |
|
datasets: |
|
- cerebras/SlimPajama-627B |
|
- EleutherAI/pile |
|
- bigcode/starcoderdata |
|
- oscar-corpus/OSCAR-2301 |
|
--- |
|
|
|
# RWKV-5 World |
|
|
|
Use rwkv pip package 0.8.22+ for RWKV-5 inference: https://pypi.org/project/rwkv/ (pipeline = PIPELINE(model, "rwkv_vocab_v20230424") for rwkv-world models) |
|
|
|
Online 7B Demo: https://huggingface.co/spaces/BlinkDL/RWKV-Gradio-2 |
|
|
|
Online 1.5B Demo: https://huggingface.co/spaces/BlinkDL/RWKV-Gradio-1 |
|
|
|
GUI: https://github.com/josStorer/RWKV-Runner (see Releases) |
|
|
|
Convert to HF formet: https://github.com/BBuf/RWKV-World-HF-Tokenizer |
|
|
|
For developer: https://github.com/BlinkDL/ChatRWKV/blob/main/API_DEMO_CHAT.py |
|
|
|
https://github.com/BlinkDL/ChatRWKV/blob/main/RWKV_v5_demo.py |
|
|
|
How it works: https://twitter.com/BlinkDL_AI/status/1685230712247795713 |
|
|
|
https://www.rwkv.com/ |
|
|
|
## Model Description |
|
|
|
RWKV-5 trained on 100+ world languages (70% English, 15% multilang, 15% code). |
|
|
|
World = Some_Pile + Some_SlimPajama + Some_StarCoder + Some_OSCAR + All_Wikipedia + All_ChatGPT_Data_I_can_find |
|
|
|
RWKV-5 training: set --my_testing "r2r4" in latest RWKV-LM v4neo: https://github.com/BlinkDL/RWKV-LM |
|
|
|
World v1 = 0.59T tokens |
|
|
|
World v2 = 1.12T tokens |
|
|
|
Imagine what happens when we use more data :) |
|
|
|
Recommended fine-tuning format (use \n for newlines): |
|
``` |
|
User: xxxxxxxxxxxxxxx |
|
|
|
Assistant: xxxxxxxxxxxxxxx |
|
xxxxxxxxxxxxxxx |
|
xxxxxxxxxxxxxxx |
|
|
|
User: xxxxxxxxxxxxxxx |
|
xxxxxxxxxxxxxxx |
|
|
|
Assistant: xxxxxxxxxxxxxxx |
|
xxxxxxxxxxxxxxx |
|
xxxxxxxxxxxxxxx |
|
xxxxxxxxxxxxxxx |
|
``` |
|
|
|
A good chat prompt (better replace \n\n in xxx to \n, such that there will be no newlines in xxx): |
|
``` |
|
User: hi |
|
|
|
Assistant: Hi. I am your assistant and I will provide expert full response in full details. Please feel free to ask any question and I will always answer it. |
|
|
|
User: xxx |
|
|
|
Assistant: |
|
``` |
|
QA prompt (better replace \n\n in xxx to \n, such that there will be no newlines in xxx): |
|
``` |
|
Question: xxx |
|
|
|
Answer: |
|
``` |
|
and |
|
``` |
|
Instruction: xxx |
|
|
|
Input: xxx |
|
|
|
Response: |
|
``` |
|
|
|
!!! There should not be any space after your final ":" or you will upset the tokenizer and see non-English reponse !!! |
|
|
|
!!! There should not be any space after your final ":" or you will upset the tokenizer and see non-English reponse !!! |
|
|
|
!!! There should not be any space after your final ":" or you will upset the tokenizer and see non-English reponse !!! |
|
|
