XiangLi1999/Diffusion-LM
Diffusion-LM is a diffusion-model-based approach to controllable text generation that learns to denoise sequences of embeddings to produce text conditioned on classifier-guided constraints.

This repository implements a diffusion model architecture for text generation, replacing autoregressive decoding with an iterative denoising process over continuous text embeddings. The model trains on paired datasets (E2E, ROCstory) using a transformer backbone and supports controllable generation by training a classifier (e.g., syntactic parser) to guide the diffusion sampling process toward desired attributes.