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

RWKV替换模型中的RNN #242

Open hulucky1102 opened 2 months ago

hulucky1102 commented 2 months ago

你好,我想使用RWKV用于语音增强,请问如何实现RWKV替换模型中的RNN部分

BlinkDL commented 4 days ago

Hi could you share your RNN code and we can take a look.