diff --git a/extensions-builtin/sd_forge_multidiffusion/lib_multidiffusion/tiled_diffusion.py b/extensions-builtin/sd_forge_multidiffusion/lib_multidiffusion/tiled_diffusion.py index 698cb782..ef8f2a4b 100644 --- a/extensions-builtin/sd_forge_multidiffusion/lib_multidiffusion/tiled_diffusion.py +++ b/extensions-builtin/sd_forge_multidiffusion/lib_multidiffusion/tiled_diffusion.py @@ -433,10 +433,6 @@ class MultiDiffusion(AbstractDiffusion): # Background sampling (grid bbox) if self.draw_background: for batch_id, bboxes in enumerate(self.batched_bboxes): - if memory_management.processing_interrupted(): - # self.pbar.close() - return x_in - # batching & compute tiles x_tile = torch.cat([x_in[bbox.slicer] for bbox in bboxes], dim=0) # [TB, C, TH, TW] n_rep = len(bboxes) @@ -536,10 +532,6 @@ class MixtureOfDiffusers(AbstractDiffusion): # Global sampling if self.draw_background: for batch_id, bboxes in enumerate(self.batched_bboxes): # batch_id is the `Latent tile batch size` - if memory_management.processing_interrupted(): - # self.pbar.close() - return x_in - # batching x_tile_list = [] t_tile_list = []