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

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

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

..

diffusionmodules

…

distributions

…

encoders

…

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

…

sub_quadratic_attention.py

…

temporal_ae.py

…