Bagheera 99c0483b67 add skip_layers argument to SD3 transformer model class (#9880) ...

* add skip_layers argument to SD3 transformer model class

* add unit test for skip_layers in stable diffusion 3

* sd3: pipeline should support skip layer guidance

* up

---------

Co-authored-by: bghira <bghira@users.github.com>
Co-authored-by: yiyixuxu <yixu310@gmail.com>

2024-11-19 15:22:54 -05:00

..

__init__.py

[Core] move transformer scripts to transformers modules (#6747)

2024-01-29 22:28:28 +05:30

test_models_dit_transformer2d.py

[Tests] clean up and refactor gradient checkpointing tests (#9494)

2024-10-31 18:24:19 +05:30

test_models_pixart_transformer2d.py

[Tests] clean up and refactor gradient checkpointing tests (#9494)

2024-10-31 18:24:19 +05:30

test_models_prior.py

Remove unnecessary lines (#8569)

2024-07-08 10:42:02 -10:00

test_models_transformer_allegro.py

[Tests] clean up and refactor gradient checkpointing tests (#9494)

2024-10-31 18:24:19 +05:30

test_models_transformer_aura_flow.py

[Tests] clean up and refactor gradient checkpointing tests (#9494)

2024-10-31 18:24:19 +05:30

test_models_transformer_cogvideox.py

CogVideoX 1.5 (#9877)

2024-11-19 00:56:34 +05:30

test_models_transformer_cogview3plus.py

[Tests] clean up and refactor gradient checkpointing tests (#9494)

2024-10-31 18:24:19 +05:30

test_models_transformer_flux.py

[Tests] clean up and refactor gradient checkpointing tests (#9494)

2024-10-31 18:24:19 +05:30

test_models_transformer_hunyuan_dit.py

[Tests] Improve transformers model test suite coverage - Hunyuan DiT (#8916)

2024-08-06 12:59:30 +05:30

test_models_transformer_latte.py

[Tests] clean up and refactor gradient checkpointing tests (#9494)

2024-10-31 18:24:19 +05:30

test_models_transformer_lumina.py

[CI] More Fast GPU Test Fixes (#9346)

2024-09-03 13:22:38 +05:30

test_models_transformer_mochi.py

[core] Mochi T2V (#9769)

2024-11-05 20:33:41 +05:30

test_models_transformer_sd3.py

add skip_layers argument to SD3 transformer model class (#9880)

2024-11-19 15:22:54 -05:00

test_models_transformer_temporal.py

[Tests] Improve transformers model test suite coverage - Temporal Transformer (#8932)

2024-07-23 15:36:30 +05:30