Can RWKV-v4 handle summarization tasks?

BlinkDL / RWKV-LM

RWKV is an RNN with transformer-level LLM performance. It can be directly trained like a GPT (parallelizable). So it's combining the best of RNN and transformer - great performance, fast inference, saves VRAM, fast training, "infinite" ctx_len, and free sentence embedding.

Apache License 2.0

11.99k stars 825 forks source link

Can RWKV-v4 handle summarization tasks? #227

Open zzczzc20 opened 4 months ago

zzczzc20 commented 4 months ago

Dear developers, I want to use ROUGE to benchmark RWKV-v4 in some summarization tasks. Is it suitable?

model: RWKV-4-Pile-1B5-20220903-8040

BlinkDL commented 4 months ago

Use rwkv6 model https://huggingface.co/BlinkDL/rwkv-6-world/blob/main/RWKV-x060-World-1B6-v2-20240208-ctx4096.pth