search

tanelp / tiny-diffusion

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

Feature/cuda #2

Open TanguyUrvoy opened 1 year ago

TanguyUrvoy commented 1 year ago

Hi,

Thanks for this great tiny tutorial. It's very helpful. I added cuda support and reduced the number of epochs to 150 because I am not patient enough. Default behavious is to use cpu.

Tanguy