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
12.47k stars 847 forks source link

Please add rocm support #247

Open Wintoplay opened 3 months ago

Wintoplay commented 3 months ago

Please consider adding rocm support for amd gpu (consumer 7900xtx and MI series)

BlinkDL commented 3 weeks ago

check https://pypi.org/project/rwkv-fla/