├── CONTRIBUTING.md ├── LICENSE ├── MaskGIT_demo.ipynb ├── README.md ├── imgs ├── class-conditional-teaser-small.png └── teaser.png ├── maskgit ├── configs │ ├── base_config.py │ ├── maskgit_class_cond_config.py │ └── vqgan_config.py ├── inference.py ├── libml │ ├── losses.py │ ├── mask_schedule.py │ └── parallel_decode.py ├── nets │ ├── discriminator.py │ ├── layers.py │ ├── maskgit_transformer.py │ └── vqgan_tokenizer.py └── utils.py └── requirements.txt /CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/maskgit/HEAD/CONTRIBUTING.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/maskgit/HEAD/LICENSE -------------------------------------------------------------------------------- /MaskGIT_demo.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/maskgit/HEAD/MaskGIT_demo.ipynb -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/maskgit/HEAD/README.md -------------------------------------------------------------------------------- /imgs/class-conditional-teaser-small.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/maskgit/HEAD/imgs/class-conditional-teaser-small.png -------------------------------------------------------------------------------- /imgs/teaser.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/maskgit/HEAD/imgs/teaser.png -------------------------------------------------------------------------------- /maskgit/configs/base_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/maskgit/HEAD/maskgit/configs/base_config.py -------------------------------------------------------------------------------- /maskgit/configs/maskgit_class_cond_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/maskgit/HEAD/maskgit/configs/maskgit_class_cond_config.py -------------------------------------------------------------------------------- /maskgit/configs/vqgan_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/maskgit/HEAD/maskgit/configs/vqgan_config.py -------------------------------------------------------------------------------- /maskgit/inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/maskgit/HEAD/maskgit/inference.py -------------------------------------------------------------------------------- /maskgit/libml/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/maskgit/HEAD/maskgit/libml/losses.py -------------------------------------------------------------------------------- /maskgit/libml/mask_schedule.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/maskgit/HEAD/maskgit/libml/mask_schedule.py -------------------------------------------------------------------------------- /maskgit/libml/parallel_decode.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/maskgit/HEAD/maskgit/libml/parallel_decode.py -------------------------------------------------------------------------------- /maskgit/nets/discriminator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/maskgit/HEAD/maskgit/nets/discriminator.py -------------------------------------------------------------------------------- /maskgit/nets/layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/maskgit/HEAD/maskgit/nets/layers.py -------------------------------------------------------------------------------- /maskgit/nets/maskgit_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/maskgit/HEAD/maskgit/nets/maskgit_transformer.py -------------------------------------------------------------------------------- /maskgit/nets/vqgan_tokenizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/maskgit/HEAD/maskgit/nets/vqgan_tokenizer.py -------------------------------------------------------------------------------- /maskgit/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/maskgit/HEAD/maskgit/utils.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/maskgit/HEAD/requirements.txt --------------------------------------------------------------------------------