tdrussell 6a26328842 Support fp16 for Cosmos-Predict2 and Anima (#12249)		2026-02-06 20:12:15 -05:00
..
cosmos_tokenizer	Replace manual x * sigmoid(x) with torch silu in VAE nonlinearity (#9057)	2025-07-30 19:25:56 -04:00
blocks.py	Enable Runtime Selection of Attention Functions (#9639)	2025-09-12 18:07:38 -04:00
model.py	Enable Runtime Selection of Attention Functions (#9639)	2025-09-12 18:07:38 -04:00
position_embedding.py	Uncap cosmos predict2 res and fix mem estimation. (#8518)	2025-06-13 07:30:18 -04:00
predict2.py	Support fp16 for Cosmos-Predict2 and Anima (#12249)	2026-02-06 20:12:15 -05:00
vae.py	Fix cosmos VAE failing with videos longer than 121 frames.	2025-01-16 16:30:06 -05:00