Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-08-27 07:31:01 +02:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #40 from space-nuko/unipc

Browse Source 
integrate unipc sampler
This commit is contained in:
Vladimir Mandic
2023-03-10 09:57:24 -05:00
committed by GitHub
parent b48d05ca5a 716a69237c
commit 2dabba9190
10 changed files with 1009 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/processing.py
+3 -1
View File
@@ -896,7 +896,9 @@ class StableDiffusionProcessingTxt2Img(StableDiffusionProcessing):
shared.state.nextjob()
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 in ['PLMS', 'UniPC']: # PLMS/UniPC do not support img2img so we just silently switch to DDIM
img2img_sampler_name = '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]