josc146 d7abe5f0d1 add pre-compiled beta cuda kernel (rwkv-beta==0.8.5, 40%+ faster for fp16) (thanks to #180, pre-compiled kernel of RTX 40 Series will be included later)

2023-09-18 23:02:49 +08:00

..

cuda

upgrade cuda-beta

2023-09-15 16:30:11 +08:00

model.py

add pre-compiled beta cuda kernel (rwkv-beta==0.8.5, 40%+ faster for fp16) (thanks to #180, pre-compiled kernel of RTX 40 Series will be included later)

2023-09-18 23:02:49 +08:00

wkv_cuda.pyd

add pre-compiled beta cuda kernel (rwkv-beta==0.8.5, 40%+ faster for fp16) (thanks to #180, pre-compiled kernel of RTX 40 Series will be included later)

2023-09-18 23:02:49 +08:00