OPENGIT/stable-diffusion-webui
1
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2025-01-06 07:05:06 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #15380 from light-and-ray/interrupt_upscale

Browse Source 
interrupt upscale
This commit is contained in:
AUTOMATIC1111 2024-03-30 07:34:34 +03:00 committed by GitHub
commit 642bca4c3d
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
modules/upscaler.py
View File

@ -60,6 +60,9 @@ class Upscaler:
if img.width >= dest_w and img.height >= dest_h: if img.width >= dest_w and img.height >= dest_h:
break break
if shared.state.interrupted:
break
shape = (img.width, img.height) shape = (img.width, img.height)
img = self.do_upscale(img, selected_model) img = self.do_upscale(img, selected_model)

2
modules/upscaler_utils.py
View File

@ -69,6 +69,8 @@ def upscale_with_model(
for y, h, row in grid.tiles: for y, h, row in grid.tiles:
newrow = [] newrow = []
for x, w, tile in row: for x, w, tile in row:
if shared.state.interrupted:
return img
output = upscale_pil_patch(model, tile) output = upscale_pil_patch(model, tile)
scale_factor = output.width // tile.width scale_factor = output.width // tile.width
newrow.append([x * scale_factor, w * scale_factor, output]) newrow.append([x * scale_factor, w * scale_factor, output])