fix rwkv infer bugs

aky15

2023-11-01 4e0404e04ed890717ead276e52c927a820326ec1

1
2
3
4
5
6
7
8
9
first    one
second    two
third    three
fourth    four
fifth    five
sixth    sixth
seventh    seven
eighth    eight
ninth    nine