mrq/stable-diffusion-webui
1
1
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Include PLMS in adjust steps as it also can fail in the same way

Browse Source
This commit is contained in:
Martin Cairns 2022-10-30 11:04:27 +00:00
parent de1dc0d279
commit 34c86c12b0
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
modules/sd_samplers.py
View File

@ -208,7 +208,7 @@ class VanillaStableDiffusionSampler:
def adjust_steps_if_invalid(self, p, num_steps): def adjust_steps_if_invalid(self, p, num_steps):
if self.config.name == 'DDIM' and p.ddim_discretize == 'uniform': if (self.config.name == 'DDIM' and p.ddim_discretize == 'uniform') or (self.config.name == 'PLMS'):
valid_step = 999 / (1000 // num_steps) valid_step = 999 / (1000 // num_steps)
if valid_step == floor(valid_step): if valid_step == floor(valid_step):
return int(valid_step) + 1 return int(valid_step) + 1