This commit is contained in:
layerdiffusion
2024-08-21 22:08:07 -07:00
parent 0d8eb4c5ba
commit c1614e08c6

View File

@@ -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 = []