Kacper Michajłow 9519e2d49d Revert "Disable pytorch attention in VAE for AMD." ... It causes crashes even without pytorch attention for big sizes, and for resonable sizes it is significantly faster. This reverts commit 1cd6cd6080.		2025-10-21 12:48:34 +02:00
..
diffusionmodules	Revert "Disable pytorch attention in VAE for AMD."	2025-10-21 12:48:34 +02: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