UniPC does not support img2img (for now)

This commit is contained in:
space-nuko
2023-02-10 05:00:09 -08:00
parent 21880eb9e5
commit c88dcc20d4
3 changed files with 3 additions and 3 deletions

View File

@@ -884,7 +884,7 @@ class StableDiffusionProcessingTxt2Img(StableDiffusionProcessing):
shared.state.nextjob()
img2img_sampler_name = 'DDIM' # PLMS does not support img2img so we just silently switch ot DDIM
img2img_sampler_name = 'DDIM' # PLMS/UniPC does not support img2img so we just silently switch ot DDIM
self.sampler = sd_samplers.create_sampler(img2img_sampler_name, self.sd_model)
samples = samples[:, :, self.truncate_y//2:samples.shape[2]-(self.truncate_y+1)//2, self.truncate_x//2:samples.shape[3]-(self.truncate_x+1)//2]