search

tanelp / tiny-diffusion

A minimal PyTorch implementation of probabilistic diffusion models for 2D datasets.
628 stars 52 forks source link

Add edm.py #6

Open yuanzhi-zhu opened 1 year ago

yuanzhi-zhu commented 1 year ago

Hi,

Thanks for this inspiring tiny implementation.

I added another tiny implementation of https://github.com/NVlabs/edm

Best, Yuanzhi