From b9f6c9aa50f079e58fbb9858337d8564a3056926 Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Tue, 2 Dec 2025 00:45:22 +0200
Subject: [PATCH] Update __init__.py

---
 wanvideo/schedulers/__init__.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/wanvideo/schedulers/__init__.py b/wanvideo/schedulers/__init__.py
index 6ee36b2..00e88f9 100644
--- a/wanvideo/schedulers/__init__.py
+++ b/wanvideo/schedulers/__init__.py
@@ -31,12 +31,12 @@ scheduler_list = [
     "multitalk",
     "sa_ode_stable",
     "rcm",
-    "vitb_unipc",
+    "vibt_unipc",
 ]
 
 def get_scheduler(scheduler, steps, start_step, end_step, shift, device, transformer_dim=5120, flowedit_args=None, denoise_strength=1.0, sigmas=None, log_timesteps=False, **kwargs):
     timesteps = None
-    if scheduler == 'vitb_unipc':
+    if scheduler == 'vibt_unipc':
         sample_scheduler = ViBTScheduler()
         sample_scheduler.set_parameters(shift=shift)
         sample_scheduler.set_timesteps(steps, device=device)