This repository provides a PyTorch implementation of Google’s Parti, a pure attention-based text-to-image neural network. It includes working training code for ViT VQGan VAE used as the image tokenizer, and supports generation of images from text prompts using transformer-based architectures. The implementation enables both training the VAE encoder-decoder and running inference with the full Parti model.