lucidrains/imagen-pytorch
Implementation of Imagen, Google's Text-to-Image Neural Network, in Pytorch
From the publisher
README & documentation
Source previewRead the project’s overview, installation instructions and usage examples. The original README is the source of truth.
Implementation of Imagen, Google's Text-to-Image Neural Network that beats DALL-E2, in Pytorch. It is the new SOTA for text-to-image synthesis. Architecturally, it is actually much simpler than DALL-E2. It consists of a cascading DDPM conditioned on text embeddings from a large pretrained T5 model (attention network). It also contains dynamic clipping for improved classifier free guidance, noise level conditioning, and a memory efficient unet design. It appears neither CLIP nor prior network is needed after all. And so research continues. AI Coffee Break with Letitia | Assembly AI | Yannic Kilcher Please join if you are interested in helping out with the replication with the LAION community For simpler training, you can directly supply text strings instead of precomputing text encodings. (Although for scaling purposes, you will definitely want to precompute the textual embeddings + mask) The number of textual captions must match the batch size of the images if…
Read the full README ↗ · Preview checked 2026-10-28T06:20:52.872Z
Inside the original README — Document outline
- Imagen - Pytorch
- Shoutouts
- Install
- Usage
- Dataloader
- Multi GPU
- Command-line
- Configuration
- Training
- Sampling
- Inpainting
- Experimental
Headings are captured from the source. Links open the publisher’s document, not a locally hosted copy.
Documentation belongs to its respective authors. Reported project/model license: MIT. A listing is not a grant of reuse or training rights. Confirm the document’s own terms at the source.
What this repository does
Implementation of Imagen, Google's Text-to-Image Neural Network, in Pytorch
Repository facts
- Owner
- lucidrains
- Primary language
- Python
- Stars
- 8,429
- Forks
- 788
- Open issues + pull requests
- 105
- License
- MIT
- Archived
- No
- Default branch
- main
- Created
- 2022-05-23T22:06:13.000Z
- Last push
- 2024-10-07T13:00:53.000Z
Topics and intended use
Owner-supplied topics: artificial-intelligence, deep-learning, imagination-machine, text-to-image, text-to-video
Review the README for scope, installation, examples and limitations. We do not run repository code or certify it.
Evaluate before installing
Review licensing and dependencies, inspect recent commits and unresolved issues, and test in an isolated environment before production use. Stars and forks alone cannot answer these questions.
Source and freshness
Source: GitHub. Metadata observed 2026-11-01T13:20:54.848Z. Daily imports are snapshots, not real-time monitoring.
Popularity and source listings do not establish security, suitability, licensing rights or benchmark performance.
Related repositories
EbookFoundation/free-programming-books
donnemartin/system-design-primer