Sayak Paul 2261510bbc [Core] Add AuraFlow (#8796) ...

* add lavender flow transformer

---------

Co-authored-by: YiYi Xu <yixu310@gmail.com>

2024-07-11 08:50:19 -10:00

..

autoencoders

[Tests] fix sharding tests (#8764)

2024-07-04 08:50:59 +05:30

transformers

[Core] Add AuraFlow (#8796)

2024-07-11 08:50:19 -10:00

unets

fix loading sharded checkpoints from subfolder (#8798)

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

__init__.py

[Tests] Improve unet / vae tests (#1018)

2022-10-28 13:43:26 +02:00

test_activations.py

move activation dispatches into helper function (#3656)

2023-06-05 12:30:48 -07:00

test_attention_processor.py

adding back test_conversion_when_using_device_map (#7704)

2024-04-18 19:21:32 -10:00

test_layers_utils.py

[LoRA] use the PyTorch classes wherever needed and start depcrecation cycles (#7204)

2024-03-13 07:56:19 +05:30

test_modeling_common_flax.py

[Tests] parallelize (#3078)

2023-04-13 13:32:57 +01:00

test_modeling_common.py

[Tests] fix more sharding tests (#8797)

2024-07-09 13:09:36 +05:30