Skip to content

Commit

Permalink
Exit out of hires fix if interrupted earlier
Browse files Browse the repository at this point in the history
  • Loading branch information
catboxanon committed Aug 18, 2023
1 parent 3ce5fb8 commit 9d1d63a
Showing 1 changed file with 3 additions and 0 deletions.
3 changes: 3 additions & 0 deletions modules/processing.py
Original file line number Diff line number Diff line change
Expand Up @@ -1155,6 +1155,9 @@ def sample(self, conditioning, unconditional_conditioning, seeds, subseeds, subs
devices.torch_gc()

def sample_hr_pass(self, samples, decoded_samples, seeds, subseeds, subseed_strength, prompts):
if shared.state.interrupted:
return samples

self.is_hr_pass = True

target_width = self.hr_upscale_to_x
Expand Down

0 comments on commit 9d1d63a

Please sign in to comment.