comfyanonymous 008761166f Optimize first attention block in cosmos VAE.		2025-01-15 21:48:46 -05:00
..
diffusionmodules	Optimize first attention block in cosmos VAE.	2025-01-15 21:48:46 -05:00
distributions	…
encoders	…
attention.py	Add argument to skip the output reshaping in the attention functions.	2025-01-10 06:27:37 -05:00
ema.py	…
sub_quadratic_attention.py	…
temporal_ae.py	…