will you work on image generation as well?

[angrysword]

Thanks so much for this. if you can work out a simple image generation as well, that will be wonderful. Mingpt have this function, prefer more mature one like put some diffusion model in or any other interested.