GitLab

* karras + VE, not flexible yet

* Fix inputs incompatibility with the original unet

* Roll back sigma scaling

* Apply suggestions from code review

* Old comment

* Fix doc

* Add torch_device to the VE pipeline

* Mark the training test with slow

* Expose schedulers

* Update __init__.py
Co-authored-by: Anton Lozhkov <anton@huggingface.co>

up

* up

* change model name

* renaming

* more changes

* up

* up

* up

* save checkpoint

* finish api / naming

* finish config renaming

* rename all weights

* finish really

* big purge

* more fixes

* finish for now

* upload

* make checkpoint work

* finalize

* uP

* finish downsampling layers

* finish major refactor

* remove bugus file