From 21cdca4283e144e5bdefd201d70947d22b0f6b78 Mon Sep 17 00:00:00 2001 From: Gabriel Date: Tue, 21 May 2024 09:45:11 +0200 Subject: [PATCH] feat: added three more samplers (#120) * Added three more samplers * Minor typo --- stable_horde/horde.py | 24 ++++++++++++++++++++++++ 1 file changed, 24 insertions(+) diff --git a/stable_horde/horde.py b/stable_horde/horde.py index a9708aa..12adde1 100644 --- 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)