Commit fb274229 authored by space-nuko's avatar space-nuko
Browse files

bug fix

parent 06cb0dc9

modules/models/diffusion/uni_pc/sampler.py

+1 −1
Original line number Diff line number Diff line
@@ -70,7 +70,7 @@ class UniPCSampler(object): 
        # sampling

        C, H, W = shape

        size = (batch_size, C, H, W)

        print(f'Data shape for UniPC sampling is {size}, eta {eta}')

        print(f'Data shape for UniPC sampling is {size}')



        device = self.model.betas.device

        if x_T is None:

modules/processing.py

+3 −1
Original line number Diff line number Diff line
@@ -884,7 +884,9 @@ class StableDiffusionProcessingTxt2Img(StableDiffusionProcessing): 


        shared.state.nextjob()



        img2img_sampler_name = 'DDIM'  # PLMS/UniPC does not support img2img so we just silently switch ot DDIM

        img2img_sampler_name = self.sampler_name

        if self.sampler_name in ['PLMS', 'UniPC']:  # PLMS/UniPC do not support img2img so we just silently switch to DDIM

            img2img_sampler_name = 'DDIM'

        self.sampler = sd_samplers.create_sampler(img2img_sampler_name, self.sd_model)



        samples = samples[:, :, self.truncate_y//2:samples.shape[2]-(self.truncate_y+1)//2, self.truncate_x//2:samples.shape[3]-(self.truncate_x+1)//2]

CRA Git | Maintained and supported by SUSTech CRA and CCSE