YiYi Xu 9e9ed353a2 fix loading sharded checkpoints from subfolder (#8798) ...

* fix load sharded checkpoints from subfolder{

* style

* os.path.join

* add a small test

---------

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

2024-07-06 11:32:04 -10:00

..

__init__.py

[Refactor] harmonize the module structure for models in tests (#6738)

2024-02-01 14:23:39 +05:30

test_models_unet_1d.py

[Tests] reduce block sizes of UNet and VAE tests (#7560)

2024-04-05 10:08:32 +05:30

test_models_unet_2d_condition.py

fix loading sharded checkpoints from subfolder (#8798)

2024-07-06 11:32:04 -10:00

test_models_unet_2d_flax.py

[Refactor] harmonize the module structure for models in tests (#6738)

2024-02-01 14:23:39 +05:30

test_models_unet_2d.py

Errata - Fix typos & improve contributing page (#8572)

2024-06-24 14:13:03 +05:30

test_models_unet_3d_condition.py

[Tests] reduce block sizes of UNet and VAE tests (#7560)

2024-04-05 10:08:32 +05:30

test_models_unet_controlnetxs.py

[Tests] mark UNetControlNetXSModelTests::test_forward_no_control to be flaky (#7771)

2024-04-25 07:29:04 +05:30

test_models_unet_motion.py

Motion Model / Adapter versatility (#8301)

2024-06-27 11:11:29 +05:30

test_models_unet_spatiotemporal.py

[Tests] reduce block sizes of UNet and VAE tests (#7560)

2024-04-05 10:08:32 +05:30

test_unet_2d_blocks.py

change to 2024 in the license (#6902)

2024-02-08 08:19:31 -10:00

test_unet_blocks_common.py

change to 2024 in the license (#6902)

2024-02-08 08:19:31 -10:00