From af2a568c8e1f04269be438e4d8aa99fbea26b19e Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Sun, 6 Oct 2024 11:54:47 +0300
Subject: [PATCH] Update nodes.py

---
 nodes.py | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/nodes.py b/nodes.py
index 79301ab..1d114d7 100644
--- a/nodes.py
+++ b/nodes.py
@@ -1042,8 +1042,8 @@ class CogVideoXFunSampler:
         
         # Load Sampler
         if context_options is not None and context_options["context_schedule"] == "temporal_tiling":
-            logging.info("Temporal tiling enabled, changing scheduler to DDIM_tiled")
-            scheduler="DDIM_tiled"
+            logging.info("Temporal tiling enabled, changing scheduler to CogVideoXDDIM")
+            scheduler="CogVideoXDDIM"
         scheduler_config = pipeline["scheduler_config"]
         if scheduler in scheduler_mapping:
             noise_scheduler = scheduler_mapping[scheduler].from_config(scheduler_config)
@@ -1373,8 +1373,8 @@ class CogVideoXFunControlSampler:
         # Load Sampler
         scheduler_config = pipeline["scheduler_config"]
         if context_options is not None and context_options["context_schedule"] == "temporal_tiling":
-            logging.info("Temporal tiling enabled, changing scheduler to DDIM_tiled")
-            scheduler="DDIM_tiled"
+            logging.info("Temporal tiling enabled, changing scheduler to CogVideoXDDIM")
+            scheduler="CogVideoXDDIM"
         if scheduler in scheduler_mapping:
             noise_scheduler = scheduler_mapping[scheduler].from_config(scheduler_config)
             pipe.scheduler = noise_scheduler