Add --rwkv-cuda-on parameter, bump rwkv version

This commit is contained in:
oobabooga 2023-03-06 20:12:54 -03:00
parent eebec65075
commit 153dfeb4dd
3 changed files with 4 additions and 3 deletions

View file

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