aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorinvincibledude <>2023-01-22 00:03:16 +0300
committerinvincibledude <>2023-01-22 00:03:16 +0300
commit9e1f49c4e5b169eecbf075c49a304774dfdc6035 (patch)
treec6ded700009593085f2bbb6f25a3bd27453ab35a
parent8bec3a2aa10d3c8ab15cca51b3b498e6ba350105 (diff)
PLMS edge-case handling fix
-rw-r--r--modules/processing.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/modules/processing.py b/modules/processing.py
index 65673a9d..6e04b064 100644
--- a/modules/processing.py
+++ b/modules/processing.py
@@ -864,7 +864,7 @@ class StableDiffusionProcessingTxt2Img(StableDiffusionProcessing):
img2img_sampler_name = self.sampler_name if self.sampler_name != 'PLMS' else 'DDIM' # PLMS does not support img2img so we just silently switch ot DDIM
- if self.hr_sampler != 0 and self.hr_sampler != 'PLMS':
+ if self.hr_sampler != 0 and sd_samplers.samplers[self.hr_sampler].name != 'PLMS':
img2img_sampler_name = self.hr_sampler
self.sampler = sd_samplers.create_sampler(img2img_sampler_name, self.sd_model)