Rattus 1d53c1f8f7 model: Add temporal roll to main VAE encoder ...

If there are no attention layers, its a standard resnet and VideoConv3d
is asked for, substitute in the temporal rolling VAE algorithm. This
reduces VAE usage by the temporal dimension (can be huge VRAM savings).

2025-11-30 09:08:17 +10:00

..

diffusionmodules

model: Add temporal roll to main VAE encoder

2025-11-30 09:08:17 +10:00

distributions

Small optimizations.

2024-12-18 18:23:28 -05:00

encoders

Make unclip more deterministic.

2024-01-14 17:28:31 -05:00

attention.py

Fix depending on asserts to raise an exception in BatchedBrownianTree and Flash attn module (#9884)

2025-09-15 20:05:03 -04:00

ema.py

Initial commit.

2023-01-16 22:37:14 -05:00

sub_quadratic_attention.py

Fix pytorch warning. (#8593)

2025-06-19 11:04:52 -04:00

temporal_ae.py

Basic Hunyuan Video model support.

2024-12-16 19:35:40 -05:00