Unverified Commit d73db17e authored by AUTOMATIC1111's avatar AUTOMATIC1111 Committed by GitHub
Browse files

Merge pull request #12515 from catboxanon/fix/gc1

Clear sampler and garbage collect before decoding images to reduce VRAM
parents 127ab911 69f49c8d
Loading
Loading
Loading
Loading
+3 −0
Original line number Original line Diff line number Diff line
@@ -1195,6 +1195,9 @@ class StableDiffusionProcessingTxt2Img(StableDiffusionProcessing):


        sd_models.apply_token_merging(self.sd_model, self.get_token_merging_ratio())
        sd_models.apply_token_merging(self.sd_model, self.get_token_merging_ratio())


        self.sampler = None
        devices.torch_gc()

        decoded_samples = decode_latent_batch(self.sd_model, samples, target_device=devices.cpu, check_for_nans=True)
        decoded_samples = decode_latent_batch(self.sd_model, samples, target_device=devices.cpu, check_for_nans=True)


        self.is_hr_pass = False
        self.is_hr_pass = False