You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
972 B
972 B
min(DALL·E)
This is a minimal implementation of DALL·E Mini in both Flax and PyTorch
Setup
Run sh setup.sh to install dependencies and download pretrained models. The only required dependencies are flax and torch. In the bash script, GitHub LFS is used to download the VQGan detokenizer and the Weight & Biases python package is used to download the DALL·E Mini and DALL·E Mega transformer models. Those files can also be downloaded manually by visting the links in the bash script.
Run
Here are some examples
python image_from_text.py --seed=7 --text='alien life'
python image_from_text.py --mega --seed=4 --text='a comfy chair that looks like an avocado'
python image_from_text.py --mega --seed=100 --text='court sketch of godzilla on trial'
