12 Commits

Author SHA1 Message Date
comfyanonymous
6e64ed0da5 Small cosmos attention code refactor. (#8530) 2025-06-14 05:02:05 -04:00
comfyanonymous
e4a3500e99 Uncap cosmos predict2 res and fix mem estimation. (#8518) 2025-06-13 07:30:18 -04:00
comfyanonymous
7cfe43744c Basic initial support for cosmos predict2 text to image 2B and 14B models. (#8517) 2025-06-13 07:05:23 -04:00
comfyanonymous
76f8667328 Change cosmos and hydit models to use the native RMSNorm. (#7934) 2025-05-04 06:26:20 -04:00
comfyanonymous
d439446067 Fix some cosmos fp8 issues. 2025-01-16 17:45:37 -05:00
comfyanonymous
63af9e9d90 Fix cosmos VAE failing with videos longer than 121 frames. 2025-01-16 16:30:06 -05:00
comfyanonymous
868a4d1551 Clean up some debug lines. 2025-01-16 04:24:39 -05:00
comfyanonymous
3423ffcf70 Lower cosmos diffusion model memory usage. 2025-01-15 23:46:42 -05:00
comfyanonymous
d20c6506b5 Lower cosmos VAE memory usage by a bit. 2025-01-15 22:57:52 -05:00
comfyanonymous
8bf2059401 Optimize first attention block in cosmos VAE. 2025-01-15 21:48:46 -05:00
comfyanonymous
21a77abd0b Remove useless code. 2025-01-13 03:52:37 -05:00
comfyanonymous
6d9f172aeb WIP support for Nvidia Cosmos 7B and 14B text to world (video) models. 2025-01-10 09:14:16 -05:00