0.6.0

prepare so that unet can work with a channel of one, and also make it…

0.5.2

save samples and models to ./results path

0.5.1

save samples and models to ./results path

0.5.0

update with new and improved cosine noise scheduler

0.4.0

new paper suggests image linear attention is more effective without q…

0.3.2

fix bug with rezero and linear attention

0.3.1

fix bug

0.3.0

make sure when sampling, batch does not exceed training batch size

0.2.4

allow for mixed precision training with fp16 flag

0.2.3

allow for mixed precision training with fp16 flag