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

Update nodes.py

Browse Source
This commit is contained in:
kijai 2024-11-12 00:58:00 +02:00
parent c8772c3aa0
commit 9681c83c1d
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

@ -842,7 +842,7 @@ class CogVideoSampler:
context_options is not None context_options is not None
), "1.0 I2V model can only do 49 frames" ), "1.0 I2V model can only do 49 frames"
if image_cond_latents is not None: if image_cond_latents is not None:
assert image_cond_latents.shape[0] == 1, "Image condition latents must be a single latent" assert image_cond_latents["samples"].shape[0] == 1, "Image condition latents must be a single latent"
assert "I2V" in pipeline.get("model_name", ""), "Image condition latents only supported for I2V models" assert "I2V" in pipeline.get("model_name", ""), "Image condition latents only supported for I2V models"
else: else:
assert "I2V" not in pipeline.get("model_name", ""), "Image condition latents required for I2V models" assert "I2V" not in pipeline.get("model_name", ""), "Image condition latents required for I2V models"