wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-04-15 04:52:31 +08:00
Code Issues Actions 13 Packages Projects Releases Wiki Activity

Add CogVideoX 1.5 geometry defaults to I2V path

Browse Source
This commit is contained in:
Talmaj Marinc 2026-04-10 19:50:16 +02:00
parent 8cf387b968
commit ff916d8896
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
comfy/supported_models.py
View File

@ -1795,6 +1795,10 @@ class CogVideoX_I2V(CogVideoX_T2V):
} }
def get_model(self, state_dict, prefix="", device=None): 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) out = model_base.CogVideoX(self, image_to_video=True, device=device)
return out return out