【文件属性】:
文件名称:DALLE-pytorch:在Pytorch中将OpenAI的DALL-E(OpenAI的文本复制到图像转换器)的实现复制
文件大小:606KB
文件格式:ZIP
更新时间:2021-02-05 09:49:19
deep-learning transformers artificial-intelligence multi-modal attention-mechanism
焦炉火中的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 ,
【文件预览】:
DALLE-pytorch-main
----.gitignore(2KB)
----README.md(6KB)
----.github()
--------workflows()
----install_deepspeed.sh(169B)
----LICENSE(1KB)
----dalle_pytorch()
--------reversible.py(5KB)
--------__init__.py(65B)
--------dalle_pytorch.py(14KB)
--------transformer.py(6KB)
----images()
--------landscape.png(591KB)
----setup.py(774B)