comfyanonymous 29596bd53f Small cosmos attention code refactor. (#8530)		2025-06-14 05:02:05 -04:00
..
cosmos_tokenizer
blocks.py	Basic initial support for cosmos predict2 text to image 2B and 14B models. (#8517)	2025-06-13 07:05:23 -04:00
model.py	Change cosmos and hydit models to use the native RMSNorm. (#7934)	2025-05-04 06:26:20 -04:00
position_embedding.py	Uncap cosmos predict2 res and fix mem estimation. (#8518)	2025-06-13 07:30:18 -04:00
predict2.py	Small cosmos attention code refactor. (#8530)	2025-06-14 05:02:05 -04:00
vae.py