From 4ce60f2176e90009a4646828490d622981398f38 Mon Sep 17 00:00:00 2001
From: lllyasviel <lyuminzhang@outlook.com>
Date: Wed, 21 Feb 2024 15:42:38 -0800
Subject: [PATCH] support SDXL-Lightning

added samplers: Euler SGMUniform, Euler A SGMUniform, DPM++ 2M SGMUniform, DPM++ 2M SDE SGMUniform

ByteDance official recommended one is Euler SGMUniform

Remember to use CFG=1.0
---
 modules_forge/forge_alter_samplers.py | 4 ++++
 modules_forge/forge_version.py        | 2 +-
 2 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/modules_forge/forge_alter_samplers.py b/modules_forge/forge_alter_samplers.py
index 1a7b728d..4e482208 100644
--- a/modules_forge/forge_alter_samplers.py
+++ b/modules_forge/forge_alter_samplers.py
@@ -38,4 +38,8 @@ samplers_data_alter = [
     sd_samplers_common.SamplerData('DPM++ 2M Turbo', build_constructor(sampler_name='dpmpp_2m', scheduler_name='turbo'), ['dpmpp_2m_turbo'], {}),
     sd_samplers_common.SamplerData('DPM++ 2M SDE Turbo', build_constructor(sampler_name='dpmpp_2m_sde', scheduler_name='turbo'), ['dpmpp_2m_sde_turbo'], {}),
     sd_samplers_common.SamplerData('LCM Karras', build_constructor(sampler_name='lcm', scheduler_name='karras'), ['lcm_karras'], {}),
+    sd_samplers_common.SamplerData('Euler SGMUniform', build_constructor(sampler_name='euler', scheduler_name='sgm_uniform'), ['euler_sgm_uniform'], {}),
+    sd_samplers_common.SamplerData('Euler A SGMUniform', build_constructor(sampler_name='euler_ancestral', scheduler_name='sgm_uniform'), ['euler_ancestral_sgm_uniform'], {}),
+    sd_samplers_common.SamplerData('DPM++ 2M SGMUniform', build_constructor(sampler_name='dpmpp_2m', scheduler_name='sgm_uniform'), ['dpmpp_2m_sgm_uniform'], {}),
+    sd_samplers_common.SamplerData('DPM++ 2M SDE SGMUniform', build_constructor(sampler_name='dpmpp_2m_sde', scheduler_name='sgm_uniform'), ['dpmpp_2m_sde_sgm_uniform'], {}),
 ]
diff --git a/modules_forge/forge_version.py b/modules_forge/forge_version.py
index f18df126..97f1cf1a 100644
--- a/modules_forge/forge_version.py
+++ b/modules_forge/forge_version.py
@@ -1 +1 @@
-version = '0.0.14v1.8.0rc'
+version = '0.0.15v1.8.0rc'