diff --git a/nodes.py b/nodes.py index 7cbc3de..14767b3 100644 --- a/nodes.py +++ b/nodes.py @@ -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"} ), } } diff --git a/pipeline_cogvideox.py b/pipeline_cogvideox.py index 276a64f..57e2172 100644 --- a/pipeline_cogvideox.py +++ b/pipeline_cogvideox.py @@ -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++