From c1614e08c62022ae0b6ab25135a9d0a610a3fda2 Mon Sep 17 00:00:00 2001 From: layerdiffusion <19834515+lllyasviel@users.noreply.github.com> Date: Wed, 21 Aug 2024 22:08:07 -0700 Subject: [PATCH] fix #1385 --- .../lib_multidiffusion/tiled_diffusion.py | 8 -------- 1 file changed, 8 deletions(-) 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 = []