Aryan a85b34e7fd [refactor] CogVideoX followups + tiled decoding support (#9150) ...

* refactor context parallel cache; update torch compile time benchmark

* add tiling support

* make style

* remove num_frames % 8 == 0 requirement

* update default num_frames to original value

* add explanations + refactor

* update torch compile example

* update docs

* update

* clean up if-statements

* address review comments

* add test for vae tiling

* update docs

* update docs

* update docstrings

* add modeling test for cogvideox transformer

* make style

2024-08-14 03:53:21 +05:30

..

autoencoders

Fix Stable Audio repository id (#9016)

2024-07-30 23:17:44 +05:30

transformers

[refactor] CogVideoX followups + tiled decoding support (#9150)

2024-08-14 03:53:21 +05:30

unets

Fix loading sharded checkpoints when we have variants (#9061)

2024-08-06 13:38:44 -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

[CI] Skip flaky download tests in PR CI (#8945)

2024-07-24 09:25:06 +05:30