madaan/minimal-text-diffusion
A minimal implementation of diffusion models for text generation, training on text corpora to generate samples.

The repository provides a stripped-down implementation of denoising diffusion probabilistic models (DDPMs) adapted for text generation. It retains essential training and sampling code from the Diffusion-LM project, removing image-related components. Users can train diffusion models on custom text corpora and generate text samples through iterative denoising. The project includes simple test datasets for quick experimentation and supports controllable generation.