GitLab

* improve comments for sde_ve scheduler, init tests

* more comments, tweaking pipelines

* timesteps --> num_training_timesteps, some comments

* merge cpu test, add m1 data

* fix scheduler tests with num_train_timesteps

* make np compatible, add tests for sde ve

* minor default variable fixes

* make style and fix-copies
Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com>

* up

* more

* uP

* make dummy test pass

* save intermediate

* p

* p

* finish

* finish

* finish

* finish

* up

* up

* finish

* uP

* save intermediate

* up

* up

* up

* finished clean up

* remove @

* uP

* finish downsampling layers

* finish major refactor

* remove bugus file

* 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