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.32k stars 838 forks source link

Fix import missing in train.py #121

Open RichardErkhov opened 1 year ago

RichardErkhov commented 1 year ago

Just added fixed import of deepspeed, which causes error later