Add top_k to RWKV
This commit is contained in:
parent
827ae51f72
commit
8660227e1b
3 changed files with 5 additions and 4 deletions
|
@ -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
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue