From b6e6bdc19575ae34c9cdb3172b048fd78688730a Mon Sep 17 00:00:00 2001
From: l0stl0rd <82827604+l0stl0rd@users.noreply.github.com>
Date: Thu, 28 Mar 2024 22:51:59 +0100
Subject: [PATCH] Update schedulers.py

---
 invokeai/backend/stable_diffusion/schedulers/schedulers.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/invokeai/backend/stable_diffusion/schedulers/schedulers.py b/invokeai/backend/stable_diffusion/schedulers/schedulers.py
index c824d94dca..cb22abd643 100644
--- a/invokeai/backend/stable_diffusion/schedulers/schedulers.py
+++ b/invokeai/backend/stable_diffusion/schedulers/schedulers.py
@@ -14,6 +14,7 @@ from diffusers import (
     LMSDiscreteScheduler,
     PNDMScheduler,
     UniPCMultistepScheduler,
+    TCDScheduler,
 )
 
 SCHEDULER_MAP = {
@@ -40,4 +41,5 @@ SCHEDULER_MAP = {
     "dpmpp_sde_k": (DPMSolverSDEScheduler, {"use_karras_sigmas": True, "noise_sampler_seed": 0}),
     "unipc": (UniPCMultistepScheduler, {"cpu_only": True}),
     "lcm": (LCMScheduler, {}),
+     "TCD": (TCDScheduler, {}),
 }