diff --git a/__init__.py b/__init__.py index ff32c08..4f92785 100644 --- a/__init__.py +++ b/__init__.py @@ -11,6 +11,7 @@ NODE_CLASS_MAPPINGS = { "CFE FLUX Guidance" : CFE_Flux_Guidance, "CFE FLUX Sampler (Pipe)" : CFE_FLUX_Pipe_Sampler, "CFE FLUX Sampler" : CFE_FLUX_Sampler, + "CFE Scheduler" : CFE_Scheduler, "CFE Sigma Sampler" : CFE_Sigma_Sampler, "CFE Sigma Sampler Strings" : CFE_Sigma_Sampler_Strings, } @@ -22,6 +23,7 @@ NODE_DISPLAY_NAME_MAPPINGS = { "CFE_Flux_Guidance":"CFE Flux Guidance", "CFE_FLUX_Pipe_Sampler":"CFE Flux Sampler (Pipe)", "CFE_FLUX_Sampler":"CFE Flux Sampler", + "CFE_Scheduler":"CFE Scheduler", "CFE_Sigma_Sampler":"CFE Sigma Sampler", "CFE_Sigma_Sampler_Strings":"CFE Sigma Sampler Strings", }