diff --git a/requirements.txt b/requirements.txt
index 4450e0d..a9599bb 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -3,6 +3,6 @@ bitsandbytes==0.37.0
 flexgen==0.1.6
 gradio==3.18.0
 numpy
-rwkv
+rwkv==0.0.6
 safetensors==0.2.8
 git+https://github.com/huggingface/transformers