wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-04-25 01:42:36 +08:00
Code Issues Actions 32 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 9904f4d73f
commit cee57f6827
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

@ -1827,6 +1827,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