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

3 years ago · 34c86c12b0
parent de1dc0d279
commit 34c86c12b0
1 changed files with 1 additions and 1 deletions
--- a/modules/sd_samplers.py
+++ b/modules/sd_samplers.py
@ -208,7 +208,7 @@ class VanillaStableDiffusionSampler:


    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)
            if valid_step == floor(valid_step):
                return int(valid_step) + 1