yuunnn-w / RWKV_Pytorch

This is an inference framework for the RWKV large language model implemented purely in native PyTorch. The official native implementation is overly complex and lacks extensibility. Let's join the flexible PyTorch ecosystem and open-source it together!
GNU General Public License v3.0
112 stars 7 forks source link

修复采样函数再topp=0时候的随机性问题,同时修复数值稳定性和计算速度 #23

Closed uniartisan closed 5 months ago