Correctly handle the case where w0 is passed by kwargs in BatchedBrownianTree |
||
|---|---|---|
| .. | ||
| diffusionmodules | ||
| distributions | ||
| encoders | ||
| attention.py | ||
| ema.py | ||
| sub_quadratic_attention.py | ||
| temporal_ae.py | ||
Correctly handle the case where w0 is passed by kwargs in BatchedBrownianTree |
||
|---|---|---|
| .. | ||
| diffusionmodules | ||
| distributions | ||
| encoders | ||
| attention.py | ||
| ema.py | ||
| sub_quadratic_attention.py | ||
| temporal_ae.py | ||