diff options
| author | space-nuko <24979496+space-nuko@users.noreply.github.com> | 2023-03-11 00:56:14 +0000 |
|---|---|---|
| committer | space-nuko <24979496+space-nuko@users.noreply.github.com> | 2023-03-11 00:56:14 +0000 |
| commit | 5fef67f6ee949a61826a3a043ea8610fd89fc371 (patch) | |
| tree | 4c9efccb08f591e3a27b7dab6e39ce4adab17bab /modules/sd_samplers_compvis.py | |
| parent | ac38ad7e60bb0ff3194536a72dd1259edad0b30a (diff) | |
| download | stable-diffusion-webui-gfx803-5fef67f6ee949a61826a3a043ea8610fd89fc371.tar.gz stable-diffusion-webui-gfx803-5fef67f6ee949a61826a3a043ea8610fd89fc371.tar.bz2 stable-diffusion-webui-gfx803-5fef67f6ee949a61826a3a043ea8610fd89fc371.zip | |
Requested changes
Diffstat (limited to 'modules/sd_samplers_compvis.py')
| -rw-r--r-- | modules/sd_samplers_compvis.py | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/modules/sd_samplers_compvis.py b/modules/sd_samplers_compvis.py index ad39ab2b..7d07c4a5 100644 --- a/modules/sd_samplers_compvis.py +++ b/modules/sd_samplers_compvis.py @@ -140,10 +140,12 @@ class VanillaStableDiffusionSampler: def adjust_steps_if_invalid(self, p, num_steps):
if ((self.config.name == 'DDIM') and p.ddim_discretize == 'uniform') or (self.config.name == 'PLMS') or (self.config.name == 'UniPC'):
+ if self.config.name == 'UniPC' and num_steps < shared.opts.uni_pc_order:
+ num_steps = shared.opts.uni_pc_order
valid_step = 999 / (1000 // num_steps)
if valid_step == math.floor(valid_step):
return int(valid_step) + 1
-
+
return num_steps
def sample_img2img(self, p, x, noise, conditioning, unconditional_conditioning, steps=None, image_conditioning=None):
|