Mikko Tukiainen 8ad68c1393 Add missing MochiEncoder3D.gradient_checkpointing attribute (#11146) ...

* Add missing 'gradient_checkpointing = False' attr

* Add (limited) tests for Mochi autoencoder

* Apply style fixes

* pass 'conv_cache' as arg instead of kwarg

---------

Co-authored-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com>

2025-04-06 02:46:45 +05:30

..

autoencoders

Add missing MochiEncoder3D.gradient_checkpointing attribute (#11146)

2025-04-06 02:46:45 +05:30

transformers

New HunyuanVideo-I2V (#11066)

2025-03-24 21:18:40 +05:30

unets

[tests] make tests device-agnostic (part 4) (#10508)

2025-03-04 08:26:06 +00: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

[Tests] Fix more tests sayak (#10359)

2024-12-23 19:11:21 +05:30

test_layers_utils.py

Support bfloat16 for Upsample2D (#9480)

2024-10-01 16:08:12 -10:00

test_modeling_common_flax.py

[Tests] parallelize (#3078)

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

test_modeling_common.py

[tests] make cuda only tests device-agnostic (#11058)

2025-03-20 10:12:35 +00:00