GitLab

add cross_attention_dim as an argument

* [Vae and AutoencoderKL clean]

* save intermediate finished work

* more progress

* more progress

* finish modeling code

* save intermediate

* finish

* Correct tests

* up

* change model name

* renaming

* more changes

* up

* up

* up

* save checkpoint

* finish api / naming

* finish config renaming

* rename all weights

* finish really

* Add a decorator for register_to_config

* All models and test

* big purge

* more fixes

* finish for now

* added kwargs for easier intialisation of random model

* initial commit for conversion script

* current debug script

* update

* Update

* done

* add updated debug conversion script

* style

* clean conversion script

* upload

* make checkpoint work

* finalize

* 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

* 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