DALLE pytorch:在Pytorch中将OpenAI的DALL E(OpenAI的文本复制到图像转换器)的实现复制 源码

焦炉火中的DALL-E 在实现/复制 (OpenAI的文本到图像转换器)。 它还将包含用于对世代进行排名。 , 和正在开发 ! 如果您想了解DALL-E在TPU上的培训,请帮助他们。 在复制此内容之前,我们可以适应“或“ 状态 设法在仅2000张风景图像的数据集上训练了一个小的6层DALL-E! (2048个视觉标记) 安装 $ pip install dalle-pytorch 用法 火车VAE import torch from dalle_pytorch import DiscreteVAE vae = DiscreteVAE ( image_size = 256 ,