comfyanonymous 29596bd53f Small cosmos attention code refactor. (#8530)

2025-06-14 05:02:05 -04:00

..

cosmos_tokenizer

Lower cosmos VAE memory usage by a bit.

2025-01-15 22:57:52 -05:00

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

Fix cosmos VAE failing with videos longer than 121 frames.

2025-01-16 16:30:06 -05:00