Merge pull request #12515 from catboxanon/fix/gc1

Clear sampler and garbage collect before decoding images to reduce VRAM
AUTOMATIC1111 · Aug 13, 2023 · d73db17 · d73db17
2 parents 127ab91 + 69f49c8
commit d73db17
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/modules/processing.py b/modules/processing.py
@@ -1195,6 +1195,9 @@ def save_intermediate(image, index):
 
         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)
 
         self.is_hr_pass = False