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

remove prints

Browse Source
This commit is contained in:
kijai 2024-09-23 14:40:41 +03:00
parent d0d7308da5
commit 81adb0220b
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
nodes.py
View File

@ -188,7 +188,7 @@ class CogVideoTransformerEdit:
@classmethod
def INPUT_TYPES(s):
return {"required": {
"remove_blocks": ("STRING", {"default": "40", "multiline": True, "tooltip": "Comma separated list of block indices to remove, 5b blocks: 0-41, 2b model blocks 0-29"} ),
"remove_blocks": ("STRING", {"default": "15, 25, 37", "multiline": True, "tooltip": "Comma separated list of block indices to remove, 5b blocks: 0-41, 2b model blocks 0-29"} ),
}
}

4
pipeline_cogvideox.py
View File

@ -451,7 +451,7 @@ class CogVideoXPipeline(VideoSysPipeline):
latents
)
latents = latents.to(self.vae.dtype)
print("latents", latents.shape)
#print("latents", latents.shape)
# 5.5.
if image_cond_latents is not None:
@ -500,7 +500,7 @@ class CogVideoXPipeline(VideoSysPipeline):
else:
temporal_tiling = False
print("Temporal tiling disabled")
print("latents.shape", latents.shape)
#print("latents.shape", latents.shape)
with self.progress_bar(total=num_inference_steps) as progress_bar:
old_pred_original_sample = None # for DPM-solver++