Commit 34c86c12 authored by Martin Cairns's avatar Martin Cairns
Browse files

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

parent de1dc0d2

modules/sd_samplers.py

+1 −1
Original line number Diff line number Diff line
@@ -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

CRA Git | Maintained and supported by SUSTech CRA and CCSE