fix T2V

2024-11-20 10:17:11 +02:00
parent de7e069286
commit ce329e0dce
1 changed files with 2 additions and 2 deletions
@@ -713,8 +713,8 @@ class CogVideoSampler:
                freenoise=context_options["freenoise"] if context_options is not None else None,
                controlnet=controlnet,
                tora=tora_trajectory if tora_trajectory is not None else None,
-                image_cond_start_percent=image_cond_start_percent,
-                image_cond_end_percent=image_cond_end_percent
+                image_cond_start_percent=image_cond_start_percent if image_cond_latents is not None else 0.0,
+                image_cond_end_percent=image_cond_end_percent if image_cond_latents is not None else 1.0,
            )
        if not model["cpu_offloading"] and model["manual_offloading"]:
            pipe.transformer.to(offload_device)