tanelp / tiny-diffusion

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

appreciation for this great mini implementation of diffusion and some question #8

Open ShellCodingGH opened 5 months ago

ShellCodingGH commented 5 months ago

Hi~

Good work~ This is such a genius implementation of diffusion. It's very impressive to see that DDPM is implemented on such a small number of codes~ Really appreciate your effort and intelligence poured into this repo^^ If it would be convenient, may I ask what are the lines of code/area that are the mini tricks to help implement this DDPM model in a small number of codes, that is different from the original DDPM model

Thanks a lot~