├── .gitattributes ├── .gitignore ├── LICENSE ├── README.md ├── image_from_text.py ├── min-dalle-jax-pmap.ipynb ├── min-dalle-jax.ipynb ├── min_dalle_flax ├── load_params.py ├── min_dalle_flax.py ├── models │ ├── dalle_bart_decoder_flax.py │ ├── dalle_bart_encoder_flax.py │ └── vqgan_detokenizer.py └── text_tokenizer.py ├── requirements.txt └── setup.sh /.gitattributes: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kuprel/min-dalle-flax/HEAD/.gitattributes -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kuprel/min-dalle-flax/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kuprel/min-dalle-flax/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kuprel/min-dalle-flax/HEAD/README.md -------------------------------------------------------------------------------- /image_from_text.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kuprel/min-dalle-flax/HEAD/image_from_text.py -------------------------------------------------------------------------------- /min-dalle-jax-pmap.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kuprel/min-dalle-flax/HEAD/min-dalle-jax-pmap.ipynb -------------------------------------------------------------------------------- /min-dalle-jax.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kuprel/min-dalle-flax/HEAD/min-dalle-jax.ipynb -------------------------------------------------------------------------------- /min_dalle_flax/load_params.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kuprel/min-dalle-flax/HEAD/min_dalle_flax/load_params.py -------------------------------------------------------------------------------- /min_dalle_flax/min_dalle_flax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kuprel/min-dalle-flax/HEAD/min_dalle_flax/min_dalle_flax.py -------------------------------------------------------------------------------- /min_dalle_flax/models/dalle_bart_decoder_flax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kuprel/min-dalle-flax/HEAD/min_dalle_flax/models/dalle_bart_decoder_flax.py -------------------------------------------------------------------------------- /min_dalle_flax/models/dalle_bart_encoder_flax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kuprel/min-dalle-flax/HEAD/min_dalle_flax/models/dalle_bart_encoder_flax.py -------------------------------------------------------------------------------- /min_dalle_flax/models/vqgan_detokenizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kuprel/min-dalle-flax/HEAD/min_dalle_flax/models/vqgan_detokenizer.py -------------------------------------------------------------------------------- /min_dalle_flax/text_tokenizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kuprel/min-dalle-flax/HEAD/min_dalle_flax/text_tokenizer.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | flax 2 | torch 3 | wandb -------------------------------------------------------------------------------- /setup.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kuprel/min-dalle-flax/HEAD/setup.sh --------------------------------------------------------------------------------