comfyanonymous 25683b5b02 Lower cosmos diffusion model memory usage.

2025-01-15 23:46:42 -05:00

..

cosmos_tokenizer

Lower cosmos VAE memory usage by a bit.

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

blocks.py

Lower cosmos diffusion model memory usage.

2025-01-15 23:46:42 -05:00

model.py

Lower cosmos diffusion model memory usage.

2025-01-15 23:46:42 -05:00

position_embedding.py

Lower cosmos diffusion model memory usage.

2025-01-15 23:46:42 -05:00

vae.py

WIP support for Nvidia Cosmos 7B and 14B text to world (video) models.

2025-01-10 09:14:16 -05:00