GitLab

* upload files

* finish

Finish

* Catch unused params in DDP

* Fix proj_out, add test

ema model stepping done automatically now

* update mid block

* finish mid block

* refactor fir up/down sample

* remove variance scaling

* remove variance scaling from unet sde

* refactor Linear

* style

* actually remove variance scaling

* add back upsample_2d, downsample_2d

* style

* fix FirUpsample2D

* make unet rl work

* uploaad files / code

* upload files

* make style correct

* finish

Remove unnecessary functions / classes