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.05k stars 827 forks source link

Adds support for PyTorch Lightning 2.0 #178

Closed fluxlinkage closed 10 months ago

fluxlinkage commented 11 months ago

增加(基本的)PyTorch Lightning 2.0支持。 Hopper和Ada Lovelace架构的GPU使用这个可以直接训练。