Converted code to ColossalAI Libary, it runs 1.65 times faster on 4x4 grid

kuprel / min-dalle

min(DALL·E) is a fast, minimal port of DALL·E Mini to PyTorch

MIT License

3.48k stars 255 forks source link

Closed puffy310 closed 2 years ago

puffy310 commented 2 years ago

Oops it already runs at this speed gotta make some more improvements I'm closing this for now.