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

运行报错 #195

Closed surviveMiao closed 8 months ago

surviveMiao commented 9 months ago

在运行chat.py文件时,报出如下错误: image python 3.10 cuda 12.1 torch 2.0

BlinkDL commented 8 months ago

pip install rwkv --upgrade

BlinkDL commented 8 months ago

fixed in rwkv pip package 0.8.20

surviveMiao commented 8 months ago

谢谢!