update schedulers

Signed-off-by: vladmandic <mandic00@live.com>
This commit is contained in:
vladmandic
2026-02-03 13:12:10 +01:00
parent ced81d630d
commit 4f7555d9e1
3 changed files with 4 additions and 4 deletions
+2 -2
View File
@@ -79,7 +79,7 @@ try:
RESMultistepScheduler,
RESSinglestepSDEScheduler,
RiemannianFlowScheduler,
DEISMultistepScheduler,
RESDEISMultistepScheduler,
LinearRKScheduler,
LobattoScheduler,
RadauIIAScheduler,
@@ -306,7 +306,7 @@ samplers_data_diffusers = [
SamplerData('RES-Multistep 3M', lambda model: DiffusionSampler('RES-Multistep 3M', RESMultistepScheduler, model), [], {}),
SamplerData('RES-SDE 2S', lambda model: DiffusionSampler('RES-SDE 2S', RESSinglestepSDEScheduler, model), [], {}),
SamplerData('RES-SDE 3S', lambda model: DiffusionSampler('RES-SDE 3S', RESSinglestepSDEScheduler, model), [], {}),
SamplerData('DEIS-Multistep', lambda model: DiffusionSampler('DEIS Multistep', DEISMultistepScheduler, model), [], {}),
SamplerData('DEIS-Multistep', lambda model: DiffusionSampler('DEIS Multistep', RESDEISMultistepScheduler, model), [], {}),
SamplerData('DEIS-Unified 1S', lambda model: DiffusionSampler('DEIS-Unified 1S', RESUnifiedScheduler, model), [], {}),
SamplerData('DEIS-Unified 2M', lambda model: DiffusionSampler('DEIS-Unified 2M', RESUnifiedScheduler, model), [], {}),
SamplerData('Sigmoid Sigma', lambda model: DiffusionSampler('Sigmoid Sigma', CommonSigmaScheduler, model), [], {}),