├── .github ├── FUNDING.yml └── workflows │ ├── ci.yml │ └── python-publish.yml ├── .gitignore ├── LICENSE ├── MANIFEST.in ├── Makefile ├── README.md ├── configs ├── README.md ├── train_decoder_config.example.json ├── train_decoder_config.test.json └── train_prior_config.example.json ├── dalle2.png ├── dalle2_pytorch ├── __init__.py ├── cli.py ├── dalle2_pytorch.py ├── data │ └── bpe_simple_vocab_16e6.txt ├── dataloaders │ ├── README.md │ ├── __init__.py │ ├── decoder_loader.py │ ├── prior_loader.py │ └── simple_image_only_dataloader.py ├── optimizer.py ├── tokenizer.py ├── trackers.py ├── train_configs.py ├── trainer.py ├── utils.py ├── version.py ├── vqgan_vae.py └── vqgan_vae_trainer.py ├── prior.md ├── samples └── oxford.png ├── setup.py ├── test_data ├── 0.tar ├── 1.tar ├── 2.tar ├── 3.tar ├── 4.tar ├── 5.tar ├── 6.tar ├── 7.tar ├── 8.tar └── 9.tar ├── train_decoder.py └── train_diffusion_prior.py /.github/FUNDING.yml: -------------------------------------------------------------------------------- 1 | github: [nousr, Veldrovive, lucidrains] 2 | -------------------------------------------------------------------------------- /.github/workflows/ci.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucidrains/DALLE2-pytorch/HEAD/.github/workflows/ci.yml -------------------------------------------------------------------------------- /.github/workflows/python-publish.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucidrains/DALLE2-pytorch/HEAD/.github/workflows/python-publish.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucidrains/DALLE2-pytorch/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucidrains/DALLE2-pytorch/HEAD/LICENSE -------------------------------------------------------------------------------- /MANIFEST.in: -------------------------------------------------------------------------------- 1 | recursive-include dalle2_pytorch *.txt 2 | -------------------------------------------------------------------------------- /Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucidrains/DALLE2-pytorch/HEAD/Makefile -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucidrains/DALLE2-pytorch/HEAD/README.md -------------------------------------------------------------------------------- /configs/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucidrains/DALLE2-pytorch/HEAD/configs/README.md -------------------------------------------------------------------------------- /configs/train_decoder_config.example.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucidrains/DALLE2-pytorch/HEAD/configs/train_decoder_config.example.json -------------------------------------------------------------------------------- /configs/train_decoder_config.test.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucidrains/DALLE2-pytorch/HEAD/configs/train_decoder_config.test.json -------------------------------------------------------------------------------- /configs/train_prior_config.example.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucidrains/DALLE2-pytorch/HEAD/configs/train_prior_config.example.json -------------------------------------------------------------------------------- /dalle2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucidrains/DALLE2-pytorch/HEAD/dalle2.png -------------------------------------------------------------------------------- /dalle2_pytorch/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucidrains/DALLE2-pytorch/HEAD/dalle2_pytorch/__init__.py -------------------------------------------------------------------------------- /dalle2_pytorch/cli.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucidrains/DALLE2-pytorch/HEAD/dalle2_pytorch/cli.py -------------------------------------------------------------------------------- /dalle2_pytorch/dalle2_pytorch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucidrains/DALLE2-pytorch/HEAD/dalle2_pytorch/dalle2_pytorch.py -------------------------------------------------------------------------------- /dalle2_pytorch/data/bpe_simple_vocab_16e6.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucidrains/DALLE2-pytorch/HEAD/dalle2_pytorch/data/bpe_simple_vocab_16e6.txt -------------------------------------------------------------------------------- /dalle2_pytorch/dataloaders/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucidrains/DALLE2-pytorch/HEAD/dalle2_pytorch/dataloaders/README.md -------------------------------------------------------------------------------- /dalle2_pytorch/dataloaders/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucidrains/DALLE2-pytorch/HEAD/dalle2_pytorch/dataloaders/__init__.py -------------------------------------------------------------------------------- /dalle2_pytorch/dataloaders/decoder_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucidrains/DALLE2-pytorch/HEAD/dalle2_pytorch/dataloaders/decoder_loader.py -------------------------------------------------------------------------------- /dalle2_pytorch/dataloaders/prior_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucidrains/DALLE2-pytorch/HEAD/dalle2_pytorch/dataloaders/prior_loader.py -------------------------------------------------------------------------------- /dalle2_pytorch/dataloaders/simple_image_only_dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucidrains/DALLE2-pytorch/HEAD/dalle2_pytorch/dataloaders/simple_image_only_dataloader.py -------------------------------------------------------------------------------- /dalle2_pytorch/optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucidrains/DALLE2-pytorch/HEAD/dalle2_pytorch/optimizer.py -------------------------------------------------------------------------------- /dalle2_pytorch/tokenizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucidrains/DALLE2-pytorch/HEAD/dalle2_pytorch/tokenizer.py -------------------------------------------------------------------------------- /dalle2_pytorch/trackers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucidrains/DALLE2-pytorch/HEAD/dalle2_pytorch/trackers.py -------------------------------------------------------------------------------- /dalle2_pytorch/train_configs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucidrains/DALLE2-pytorch/HEAD/dalle2_pytorch/train_configs.py -------------------------------------------------------------------------------- /dalle2_pytorch/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucidrains/DALLE2-pytorch/HEAD/dalle2_pytorch/trainer.py -------------------------------------------------------------------------------- /dalle2_pytorch/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucidrains/DALLE2-pytorch/HEAD/dalle2_pytorch/utils.py -------------------------------------------------------------------------------- /dalle2_pytorch/version.py: -------------------------------------------------------------------------------- 1 | __version__ = '1.15.6' 2 | -------------------------------------------------------------------------------- /dalle2_pytorch/vqgan_vae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucidrains/DALLE2-pytorch/HEAD/dalle2_pytorch/vqgan_vae.py -------------------------------------------------------------------------------- /dalle2_pytorch/vqgan_vae_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucidrains/DALLE2-pytorch/HEAD/dalle2_pytorch/vqgan_vae_trainer.py -------------------------------------------------------------------------------- /prior.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucidrains/DALLE2-pytorch/HEAD/prior.md -------------------------------------------------------------------------------- /samples/oxford.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucidrains/DALLE2-pytorch/HEAD/samples/oxford.png -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucidrains/DALLE2-pytorch/HEAD/setup.py -------------------------------------------------------------------------------- /test_data/0.tar: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucidrains/DALLE2-pytorch/HEAD/test_data/0.tar -------------------------------------------------------------------------------- /test_data/1.tar: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucidrains/DALLE2-pytorch/HEAD/test_data/1.tar -------------------------------------------------------------------------------- /test_data/2.tar: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucidrains/DALLE2-pytorch/HEAD/test_data/2.tar -------------------------------------------------------------------------------- /test_data/3.tar: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucidrains/DALLE2-pytorch/HEAD/test_data/3.tar -------------------------------------------------------------------------------- /test_data/4.tar: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucidrains/DALLE2-pytorch/HEAD/test_data/4.tar -------------------------------------------------------------------------------- /test_data/5.tar: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucidrains/DALLE2-pytorch/HEAD/test_data/5.tar -------------------------------------------------------------------------------- /test_data/6.tar: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucidrains/DALLE2-pytorch/HEAD/test_data/6.tar -------------------------------------------------------------------------------- /test_data/7.tar: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucidrains/DALLE2-pytorch/HEAD/test_data/7.tar -------------------------------------------------------------------------------- /test_data/8.tar: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucidrains/DALLE2-pytorch/HEAD/test_data/8.tar -------------------------------------------------------------------------------- /test_data/9.tar: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucidrains/DALLE2-pytorch/HEAD/test_data/9.tar -------------------------------------------------------------------------------- /train_decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucidrains/DALLE2-pytorch/HEAD/train_decoder.py -------------------------------------------------------------------------------- /train_diffusion_prior.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucidrains/DALLE2-pytorch/HEAD/train_diffusion_prior.py --------------------------------------------------------------------------------