search

lucidrains / DALLE2-pytorch

Implementation of DALL-E 2, OpenAI's updated text-to-image synthesis neural network, in Pytorch
MIT License
11.03k stars 1.07k forks source link

Incorporating Gaussian Diffusion with Learned Variance #90

Closed lucidrains closed 2 years ago

lucidrains commented 2 years ago

https://github.com/lucidrains/denoising-diffusion-pytorch/blob/master/denoising_diffusion_pytorch/learned_gaussian_diffusion.py not sure how significant the improvements will be, but it seems from the paper they are using this for the first stage of the cascade, so we'll include it as a setting

lucidrains commented 2 years ago

ok its done https://github.com/lucidrains/DALLE2-pytorch/commit/2277b47ffda6a0952bb40c3b326c0214705b831b