search

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

bugfix #13

Closed uniartisan closed 5 months ago

uniartisan commented 5 months ago
  1. 修复代码里缺少的：stop_token = tokenizer.encode(stop)[0]
  2. 修复测试代码的错误
  3. 增加模型的保存