xinyun/ComfyUI-CogVideoXWrapper
1
0
Fork 0
mirror of https://git.datalinker.icu/kijai/ComfyUI-CogVideoXWrapper.git synced 2026-01-23 12:24:23 +08:00
Code Issues Packages Projects Releases Wiki Activity

don't break MZ loader for no reason

Browse Source
This commit is contained in:
kijai 2024-10-08 14:53:19 +03:00
parent e1a784d92b
commit d83f64aa3e
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
nodes.py
View File

@ -870,7 +870,7 @@ class CogVideoSampler:
base_path = pipeline["base_path"]
assert "fun" not in base_path.lower(), "'Fun' models not supported in 'CogVideoSampler', use the 'CogVideoXFunSampler'"
assert ("I2V" not in pipeline["model_name"] or num_frames == 49 or context_options is not None), "I2V model can only do 49 frames"
assert ("I2V" not in pipeline.get("model_name","") or num_frames == 49 or context_options is not None), "I2V model can only do 49 frames"
device = mm.get_torch_device()
offload_device = mm.unet_offload_device()