焦炉火中的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 ,
1