fix rwkv infer bugs

aky15

2023-11-01 4e0404e04ed890717ead276e52c927a820326ec1

1
2
3
4
5
6
7
8
9
10
11
12
13    1
14    2
15    3
16    4
17    5
18    6
19    7
20    8
21    9
22    10
23    11
24    12