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
118 stars 8 forks source link

添加精度选项 #26

Closed Aknifejackzhmolong closed 6 months ago

Aknifejackzhmolong commented 6 months ago

可以在配置文件中使用float32/float16等精度训练