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

comfyanonymous · Jan 10, 2025 · 2ff3104 · 2ff3104
1 parent 129d890
commit 2ff3104
Show file tree

Hide file tree

Showing 16 changed files with 3,383 additions and 4 deletions.
diff --git a/comfy/latent_formats.py b/comfy/latent_formats.py
@@ -382,3 +382,7 @@ class HunyuanVideo(LatentFormat):
     ]
 
     latent_rgb_factors_bias = [ 0.0259, -0.0192, -0.0761]
+
+class Cosmos1CV8x8x8(LatentFormat):
+    latent_channels = 16
+    latent_dimensions = 3