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.
BlinkDL
commented
2 days ago
Hello, can the WKV acceleration module be tracked/scripted by torch.onnx.export? I'm having trouble converting the.pth model to.onnx.