RWKV / rwkv-onnx Public

Notifications You must be signed in to change notification settings
Fork 3
Star 41

A converter and basic tester for rwkv onnx

41 stars 3 forks Branches Tags Activity

Notifications

Name		Name	Last commit message	Last commit date
Latest commit History 22 Commits
tokenizer		tokenizer
.gitignore		.gitignore
README.md		README.md
convert.py		convert.py
opslist.py		opslist.py
quant.py		quant.py
test.py		test.py
toort.sh		toort.sh

Repository files navigation

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

About

A converter and basic tester for rwkv onnx

Custom properties

Report repository

Releases

No releases published

Packages

No packages published

Languages