diff options
| author | Martin Cairns <4314538+MartinCairnsSQL@users.noreply.github.com> | 2022-10-30 11:04:27 +0000 |
|---|---|---|
| committer | Martin Cairns <4314538+MartinCairnsSQL@users.noreply.github.com> | 2022-10-30 11:04:27 +0000 |
| commit | 34c86c12b0a9d650d4e7c5be478bca34ad8ed048 (patch) | |
| tree | 596c1005e92397c20a1c6a22accce9417bdd1d07 | |
| parent | de1dc0d279a877d5d9f512befe30a7d7e5cf3881 (diff) | |
Include PLMS in adjust steps as it also can fail in the same way
| -rw-r--r-- | modules/sd_samplers.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/modules/sd_samplers.py b/modules/sd_samplers.py index aca014e8..8772db56 100644 --- 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
|