diff options
| author | space-nuko <24979496+space-nuko@users.noreply.github.com> | 2023-02-10 13:00:09 +0000 |
|---|---|---|
| committer | space-nuko <24979496+space-nuko@users.noreply.github.com> | 2023-02-10 13:00:54 +0000 |
| commit | c88dcc20d495dab4be2692bdff30277112dbe416 (patch) | |
| tree | 6470af624a4769c6bc676902a0e05a9222b8f898 /modules | |
| parent | 21880eb9e57b884635a07d2360831b4186afddf4 (diff) | |
| download | stable-diffusion-webui-gfx803-c88dcc20d495dab4be2692bdff30277112dbe416.tar.gz stable-diffusion-webui-gfx803-c88dcc20d495dab4be2692bdff30277112dbe416.tar.bz2 stable-diffusion-webui-gfx803-c88dcc20d495dab4be2692bdff30277112dbe416.zip | |
UniPC does not support img2img (for now)
Diffstat (limited to 'modules')
| -rw-r--r-- | modules/processing.py | 2 | ||||
| -rw-r--r-- | modules/sd_samplers.py | 2 | ||||
| -rw-r--r-- | modules/sd_samplers_compvis.py | 2 |
3 files changed, 3 insertions, 3 deletions
diff --git a/modules/processing.py b/modules/processing.py index 11e726df..b7cf5357 100644 --- a/modules/processing.py +++ b/modules/processing.py @@ -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]
diff --git a/modules/sd_samplers.py b/modules/sd_samplers.py index 28c2136f..ff361f22 100644 --- a/modules/sd_samplers.py +++ b/modules/sd_samplers.py @@ -32,7 +32,7 @@ def set_samplers(): global samplers, samplers_for_img2img
hidden = set(shared.opts.hide_samplers)
- hidden_img2img = set(shared.opts.hide_samplers + ['PLMS'])
+ hidden_img2img = set(shared.opts.hide_samplers + ['PLMS', 'UniPC'])
samplers = [x for x in all_samplers if x.name not in hidden]
samplers_for_img2img = [x for x in all_samplers if x.name not in hidden_img2img]
diff --git a/modules/sd_samplers_compvis.py b/modules/sd_samplers_compvis.py index 946079ae..ad39ab2b 100644 --- a/modules/sd_samplers_compvis.py +++ b/modules/sd_samplers_compvis.py @@ -139,7 +139,7 @@ class VanillaStableDiffusionSampler: self.nmask = p.nmask if hasattr(p, 'nmask') else None
def adjust_steps_if_invalid(self, p, num_steps):
- if ((self.config.name == 'DDIM' or self.config.name == "UniPC") and p.ddim_discretize == 'uniform') or (self.config.name == 'PLMS'):
+ if ((self.config.name == 'DDIM') and p.ddim_discretize == 'uniform') or (self.config.name == 'PLMS') or (self.config.name == 'UniPC'):
valid_step = 999 / (1000 // num_steps)
if valid_step == math.floor(valid_step):
return int(valid_step) + 1
|