blepping b273c4af3e Fix depending on asserts to raise an exception in BatchedBrownianTree and Flash attn module ...

Correctly handle the case where w0 is passed by kwargs in BatchedBrownianTree

2025-09-15 15:20:49 -06:00

..

diffusionmodules

Enable Runtime Selection of Attention Functions (#9639)

2025-09-12 18:07:38 -04: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

2025-09-15 15:20:49 -06: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