feat: added three more samplers (#120)

* Added three more samplers * Minor typo
sdwebui-w-horde · May 21, 2024 · 21cdca4 · 21cdca4
1 parent a31d331
commit 21cdca4
Showing 1 changed file with 24 additions and 0 deletions.
diff --git a/stable_horde/horde.py b/stable_horde/horde.py
@@ -261,6 +261,30 @@ def patch_sampler_names(self):
                 ["k_dpm_fast_ka"],
                 {"scheduler": "karras"},
             ),
+            SamplerData(
+                "LMS Karras",
+                lambda model, funcname="sample_lms": KDiffusionSampler(
+                    funcname, model
+                ),
+                ["k_lms_ka"],
+                {"scheduler": "karras"},
+            ),
+            SamplerData(
+                "DPM++ SDE Karras",
+                lambda model, funcname="sample_dpmpp_sde": KDiffusionSampler(
+                    funcname, model
+                ),
+                ["k_dpmpp_sde_ka"],
+                {"scheduler": "karras"},
+            ),
+            SamplerData(
+                "DPM++ 2S a Karras",
+                lambda model, funcname="sample_dpmpp_2s_ancestral": KDiffusionSampler(
+                    funcname, model
+                ),
+                ["k_dpmpp_2s_a_ka"],
+                {"scheduler": "karras"},
+            ),
         ]
         sd_samplers.samplers.extend(samplers)
         sd_samplers.samplers_for_img2img.extend(samplers)