search

lucidrains / DALLE2-pytorch

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

🤔 #200

Closed nousr closed 2 years ago

nousr commented 2 years ago

https://github.com/lucidrains/DALLE2-pytorch/blob/a59882001267caef4ccf2879dc9b7efcccb6d93a/dalle2_pytorch/dalle2_pytorch.py#L1064-L1068

nousr commented 2 years ago

@lucidrains shouldn't you either set equal to image_embed or return img somehow?

lucidrains commented 2 years ago

@nousr yes indeed! https://github.com/lucidrains/DALLE2-pytorch/commit/3dae43fa0ef2c5a29df9c275b1ebeb1a65c3bac5 :pray: