training and infering scripts for RWKV x060 model.
inspired by https://github.com/yuunnn-w/RWKV_Pytorch and https://github.com/BlinkDL/RWKV-LM
- parallel forward with attention mask
- tokenizer apply_chat_template method
-
Notifications
You must be signed in to change notification settings - Fork 0
lithiumfleet/rwkv6_simple
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
About
some code with rwkv x060 model forwarding/training...
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published