Complete set_attn_processor for prior and vae (#3796)
* relax tolerance slightly
* Add more tests
* upload readme
* upload readme
* Apply suggestions from code review
* Improve API Autoencoder KL
* finalize
* finalize tests
* finalize tests
* Apply suggestions from code review
Co-authored-by: Sayak Paul <spsayakpaul@gmail.com>
* up
---------
Co-authored-by: Sayak Paul <spsayakpaul@gmail.com>
