From 649ecc4d347089d3b91230406563d194198aef24 Mon Sep 17 00:00:00 2001
From: spezialspezial <75758219+spezialspezial@users.noreply.github.com>
Date: Sat, 3 Dec 2022 22:10:17 +0100
Subject: [PATCH] Allow k_dpmpp_2s_ancestral and k_dpmpp_2m

Tested and works so far for 'k_dpmpp_2s_ancestral' and 'k_dpmpp_2m'. NOT working for 'k_dpm_fast', 'k_dpm_adaptive' out of the box
---
 ldm/generate.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/ldm/generate.py b/ldm/generate.py
index 8d68314f386..3f2c9347617 100644
--- a/ldm/generate.py
+++ b/ldm/generate.py
@@ -986,6 +986,8 @@ def _set_sampler(self):
             self.sampler = KSampler(self.model, 'heun', device=self.device)
         elif self.sampler_name == 'k_lms':
             self.sampler = KSampler(self.model, 'lms', device=self.device)
+        elif self.sampler_name[:2] == 'k_': # and f'sample_{self.sampler_name[2:]}' in K.sampling.__dict__:
+            self.sampler = KSampler(self.model, self.sampler_name[2:], device=self.device)
         else:
             msg = f'>> Unsupported Sampler: {self.sampler_name}, Defaulting to plms'
             self.sampler = PLMSSampler(self.model, device=self.device)