Skip to content

Latest commit

 

History

History
12 lines (10 loc) · 249 Bytes

README.md

File metadata and controls

12 lines (10 loc) · 249 Bytes

rwkv-onnx

A converter and basic tester for rwkv V5 onnx

supports

  • fp16, fp32
  • onnx opsversions 15/17/18
  • file sizes > 2GB
  • cuda, cpu, tensorRT, mps

Note: converter requires protobuf 3.20.0

Install with pip install protobuf==3.20.0