Add top_k to RWKV

This commit is contained in:
oobabooga 2023-03-07 17:24:28 -03:00
parent 827ae51f72
commit 8660227e1b
3 changed files with 5 additions and 4 deletions

View file

@ -3,7 +3,7 @@ bitsandbytes==0.37.0
flexgen==0.1.7
gradio==3.18.0
numpy
rwkv==0.0.8
rwkv==0.1.0
safetensors==0.2.8
sentencepiece
git+https://github.com/oobabooga/transformers@llama_push