FreeInit (#6315)
* freeinit
* update freeinit implementation based on review
Co-Authored-By: Dhruv Nair <dhruv.nair@gmail.com>
* fix
* another fix
* refactor
* fix timesteps missing bug
* apply suggestions from review
Co-Authored-By: Dhruv Nair <dhruv.nair@gmail.com>
* add test for freeinit
* apply suggestions from review
Co-Authored-By: Dhruv Nair <dhruv.nair@gmail.com>
* refactor
* fix test
* fix tensor not on same device
* update
* remove return_intermediate_results
* fix broken freeinit test
* update animatediff docs
---------
Co-authored-by: Dhruv Nair <dhruv.nair@gmail.com>
