ChatRWKV icon indicating copy to clipboard operation
ChatRWKV copied to clipboard

How to train a ChatRWKV model on domain-specific data

Open Just4Uzi opened this issue 2 years ago • 2 comments

Hi! Thank you for your wonderful work! I want to train a QA bot on my own data, and I guess I shuold train a RWKV-LM v4 model and load this model in chat.py(by change the model path). Am I doing right? Do you have any suggestions?

Thank you very much

Just4Uzi avatar Feb 17 '23 07:02 Just4Uzi

Hi you can finetune a RWKV-v4 model using v4neo code in RWKV-LM. Read the "large corpus" part for guide.

BlinkDL avatar Feb 17 '23 09:02 BlinkDL

Now you can also use https://github.com/Blealtan/RWKV-LM-LoRA

BlinkDL avatar Mar 15 '23 20:03 BlinkDL