tests/models/unets/test_models_unet_2d_condition.py · 96399c3ec67816987f2f8ad9b899a572dd219dd8 · OpenDAS / diffusers

Fix sharding when no device_map is passed (#8531) · 96399c3e

Marc Sun authored Jun 18, 2024



* Fix sharding when no device_map is passed

* style

* add tests

* align

* add docstring

* format

---------
Co-authored-by: Sayak Paul <spsayakpaul@gmail.com>

96399c3e

test_models_unet_2d_condition.py 56.6 KB

Replace test_models_unet_2d_condition.py