Text to Image generator based on DALL-E.
Reference DALLE-Pytorch
Python 3.10.8
Microsoft Visual C++ 14.0
CUDA Toolkit 11.7
Run setup.bat file.
Training model
python train.py --image_text_folder ./dataset --taming --truncate_captions
Generate image
python generate.py --dalle_path ./dalle.pt --text "blue jacket"