From cac1f81c519d908ab7db1bf923d2cf7ece7a6411 Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Tue, 19 Nov 2024 19:42:09 +0200
Subject: [PATCH] Update pipeline_cogvideox.py

---
 pipeline_cogvideox.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/pipeline_cogvideox.py b/pipeline_cogvideox.py
index f869ce4..39c5141 100644
--- a/pipeline_cogvideox.py
+++ b/pipeline_cogvideox.py
@@ -514,8 +514,8 @@ class CogVideoXPipeline(DiffusionPipeline, CogVideoXLoraLoaderMixin):
 
         # 7. context schedule
         if context_schedule is not None:
-            if image_cond_latents is not None:
-                raise NotImplementedError("Context schedule not currently supported with image conditioning")
+            # if image_cond_latents is not None:
+            #     raise NotImplementedError("Context schedule not currently supported with image conditioning")
             logger.info(f"Context schedule enabled: {context_frames} frames, {context_stride} stride, {context_overlap} overlap")
             use_context_schedule = True
             from .context import get_context_scheduler