Sayak Paul
|
af769881d3
|
[tests] introduce VAETesterMixin to consolidate tests for slicing and tiling (#12374)
* up
* up
* up
* up
* up
* u[
* up
* up
* up
|
2025-10-17 12:02:29 +05:30 |
|
Dhruv Nair
|
7aa6af1138
|
[Refactor] Move testing utils out of src (#12238)
* update
* update
* update
* update
* update
* merge main
* Revert "merge main"
This reverts commit 65efbcead5.
|
2025-08-28 19:53:02 +05:30 |
|
Sayak Paul
|
62cce3045d
|
[chore] change to 2025 licensing for remaining (#11741)
change to 2025 licensing for remaining
|
2025-06-18 20:56:00 +05:30 |
|
Fanli Lin
|
15ad97f782
|
[tests] make cuda only tests device-agnostic (#11058)
* enable bnb on xpu
* add 2 more cases
* add missing change
* add missing change
* add one more
* enable cuda only tests on xpu
* enable big gpu cases
|
2025-03-20 10:12:35 +00:00 |
|
dg845
|
648d968cfc
|
Enable Gradient Checkpointing for UNet2DModel (New) (#7201)
* Port UNet2DModel gradient checkpointing code from #6718.
---------
Co-authored-by: Sayak Paul <spsayakpaul@gmail.com>
Co-authored-by: Vincent Neemie <92559302+VincentNeemie@users.noreply.github.com>
Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com>
Co-authored-by: Dhruv Nair <dhruv.nair@gmail.com>
Co-authored-by: hlky <hlky@hlky.ac>
|
2024-12-19 14:45:45 -10:00 |
|
Sayak Paul
|
c1926cef6b
|
[tests] refactor vae tests (#9808)
* add: autoencoderkl tests
* autoencodertiny.
* fix
* asymmetric autoencoder.
* more
* integration tests for stable audio decoder.
* consistency decoder vae tests
* remove grad check from consistency decoder.
* cog
* bye test_models_vae.py
* fix
* fix
* remove allegro
* fixes
* fixes
* fixes
---------
Co-authored-by: Dhruv Nair <dhruv.nair@gmail.com>
|
2024-12-04 15:58:36 +05:30 |
|