aboutsummaryrefslogtreecommitdiffstats
path: root/modules/sd_samplers_compvis.py
diff options
context:
space:
mode:
authorspace-nuko <24979496+space-nuko@users.noreply.github.com>2023-02-10 13:00:09 +0000
committerspace-nuko <24979496+space-nuko@users.noreply.github.com>2023-02-10 13:00:54 +0000
commitc88dcc20d495dab4be2692bdff30277112dbe416 (patch)
tree6470af624a4769c6bc676902a0e05a9222b8f898 /modules/sd_samplers_compvis.py
parent21880eb9e57b884635a07d2360831b4186afddf4 (diff)
downloadstable-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/sd_samplers_compvis.py')
-rw-r--r--modules/sd_samplers_compvis.py2
1 files changed, 1 insertions, 1 deletions
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