Commit 9d1d63af authored by catboxanon's avatar catboxanon
Browse files

Exit out of hires fix if interrupted earlier

parent 3ce5fb8e

modules/processing.py

+3 −0
Original line number Diff line number Diff line
@@ -1155,6 +1155,9 @@ class StableDiffusionProcessingTxt2Img(StableDiffusionProcessing): 
            devices.torch_gc()



    def sample_hr_pass(self, samples, decoded_samples, seeds, subseeds, subseed_strength, prompts):

        if shared.state.interrupted:

            return samples



        self.is_hr_pass = True



        target_width = self.hr_upscale_to_x

CRA Git | Maintained and supported by SUSTech CRA and CCSE