DiT-MNIST

A small class-conditional Diffusion Transformer trained on MNIST. Pick a digit and sample new handwritten-digit images.

0 9
1 16
0 5
10 100