← all repositories

madaan/minimal-text-diffusion

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

415 stars Python Language ModelsML Frameworks
minimal-text-diffusion
Velocity · 7d
+0.3
★ / day
Trend
steady
star history

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.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.