PLMS edge-case handling fix

This commit is contained in:
invincibledude 2023-01-22 00:03:16 +03:00
parent 8bec3a2aa1
commit 9e1f49c4e5

View File

@ -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 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 img2img_sampler_name = self.hr_sampler
self.sampler = sd_samplers.create_sampler(img2img_sampler_name, self.sd_model) self.sampler = sd_samplers.create_sampler(img2img_sampler_name, self.sd_model)