This repository is for sharing pre-trained OpenAI DALLE model and generating images from given texts.
All models are trained by lucidrains/DALLE-pytorch + VQGAN (Taming transformer) with different training code and BPE model.
The notebook includes
- Text to image generation
- Pre-trained CLIP reranking
3. Generate rest of image based on the given cropped image
- Install requirements
$ pip install -r requirements
- Install DeepSpeed
- Follow the instruction here and install DeepSpeed
- Download models below and save them in pretrained folder
- Check the link in Details for the model specifics