Mathias Parger 8ae8008b0d speedup hunyuan encoder causal mask generation (#10764) ...

* speedup causal mask generation

* fixing hunyuan attn mask test case

2025-02-11 16:03:15 +05:30

..

__init__.py

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

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

test_models_asymmetric_autoencoder_kl.py

[tests] refactor vae tests (#9808)

2024-12-04 15:58:36 +05:30

test_models_autoencoder_dc.py

[DC-AE] Add the official Deep Compression Autoencoder code(32x,64x,128x compression ratio); (#9708)

2024-12-07 01:01:51 +05:30

test_models_autoencoder_hunyuan_video.py

speedup hunyuan encoder causal mask generation (#10764)

2025-02-11 16:03:15 +05:30

test_models_autoencoder_kl_cogvideox.py

[tests] refactor vae tests (#9808)

2024-12-04 15:58:36 +05:30

test_models_autoencoder_kl_temporal_decoder.py

Enable Gradient Checkpointing for UNet2DModel (New) (#7201)

2024-12-19 14:45:45 -10:00

test_models_autoencoder_kl.py

Enable Gradient Checkpointing for UNet2DModel (New) (#7201)

2024-12-19 14:45:45 -10:00

test_models_autoencoder_ltx_video.py

Implement framewise encoding/decoding in LTX Video VAE (#10488)

2025-01-13 10:58:32 -10:00

test_models_autoencoder_oobleck.py

[Tests] Test layerwise casting with training (#10765)

2025-02-11 16:02:28 +05:30

test_models_autoencoder_tiny.py

[core] Layerwise Upcasting (#10347)

2025-01-22 19:49:37 +05:30

test_models_consistency_decoder_vae.py

[tests] refactor vae tests (#9808)

2024-12-04 15:58:36 +05:30

test_models_vae_flax.py

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

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

test_models_vq.py

[Tests] skip tests properly with unittest.skip() (#10527)

2025-01-11 08:46:22 +05:30

vae.py

[tests] refactor vae tests (#9808)

2024-12-04 15:58:36 +05:30