Add CogVideoX 1.5 geometry defaults to I2V path

Talmaj Marinc · Talmaj Marinc · commit d4f7131f6e30 · 2026-04-10T19:50:16.000+02:00
diff --git a/comfy/supported_models.py b/comfy/supported_models.py
@@ -1795,6 +1795,10 @@ class CogVideoX_I2V(CogVideoX_T2V):
     }
 
     def get_model(self, state_dict, prefix="", device=None):
+        if self.unet_config.get("patch_size_t") is not None:
+            self.unet_config.setdefault("sample_height", 96)
+            self.unet_config.setdefault("sample_width", 170)
+            self.unet_config.setdefault("sample_frames", 81)
         out = model_base.CogVideoX(self, image_to_video=True, device=device)
         return out
 

Original file line number	Diff line number	Diff line change
`@@ -1795,6 +1795,10 @@ class CogVideoX_I2V(CogVideoX_T2V):`
`1795`	`1795`	`}`
`1796`	`1796`
`1797`	`1797`	`def get_model(self, state_dict, prefix="", device=None):`
	`1798`	`+ if self.unet_config.get("patch_size_t") is not None:`
	`1799`	`+ self.unet_config.setdefault("sample_height", 96)`
	`1800`	`+ self.unet_config.setdefault("sample_width", 170)`
	`1801`	`+ self.unet_config.setdefault("sample_frames", 81)`
`1798`	`1802`	`out = model_base.CogVideoX(self, image_to_video=True, device=device)`
`1799`	`1803`	`return out`
`1800`	`1804`