RWKV-Red-Team
/

ARWKV-R1-7B

Text Generation

Model card Files Files and versions

zhiyuan8 commited on May 11

Commit

e48eb77

·

verified ·

1 Parent(s): 11792f5

Update README.md

Files changed (1) hide show

README.md +2 -3

README.md CHANGED Viewed

@@ -90,10 +90,9 @@ python ./convert_hf_to_gguf.py [model_dir]
 ## How to use
 ```bash
-pip3 install --upgrade transformers
-pip3 install rwkv-fla==0.7.202505051657
 ```
 ```python
 from transformers import AutoModelForCausalLM, AutoTokenizer

 ## How to use
 ```bash
+pip3 install --upgrade transformers rwkv-fla
 ```
+Before training: `export WKV_MODE=chunk`
 ```python
 from transformers import AutoModelForCausalLM, AutoTokenizer