Skip to content

Commit 9d1d63a

Browse files
committed
Exit out of hires fix if interrupted earlier
1 parent 3ce5fb8 commit 9d1d63a

1 file changed

Lines changed: 3 additions & 0 deletions

File tree

‎modules/processing.py‎

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1155,6 +1155,9 @@ def sample(self, conditioning, unconditional_conditioning, seeds, subseeds, subs
11551155
devices.torch_gc()
11561156

11571157
def sample_hr_pass(self, samples, decoded_samples, seeds, subseeds, subseed_strength, prompts):
1158+
if shared.state.interrupted:
1159+
return samples
1160+
11581161
self.is_hr_pass = True
11591162

11601163
target_width = self.hr_upscale_to_x

0 commit comments

Comments
 (0)