search

wingedsheep / transformer

MIT License
15 stars 13 forks source link

ONNX format #1

Open bitdom8 opened 1 year ago

bitdom8 commented 1 year ago

Thanks for the great repo. How do we convert to ONNX format?

wingedsheep commented 1 year ago

Glad you like it. I created this repository for a blogpost https://wingedsheep.com/building-a-language-model/ so it's a very simple implementation.

See https://pytorch.org/tutorials/advanced/super_resolution_with_onnxruntime.html for converting to ONXX format.

bitdom8 commented 1 year ago

Tried even asked chatgpt but couldn't make it work