michael/stable-diffusion-webui
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Also make SwinIR skippable

Browse source
This commit is contained in:
catboxanon 2023-01-23 22:00:27 -05:00 committed by GitHub
parent f993525820
commit 3c47b05036
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
extensions-builtin/SwinIR/scripts/swinir_model.py
View file

@ -145,11 +145,11 @@ def inference(img, model, tile, tile_overlap, window_size, scale):
with tqdm(total=len(h_idx_list) * len(w_idx_list), desc="SwinIR tiles") as pbar: with tqdm(total=len(h_idx_list) * len(w_idx_list), desc="SwinIR tiles") as pbar:
for h_idx in h_idx_list: for h_idx in h_idx_list:
if state.interrupted: if state.interrupted or state.skipped:
break break
for w_idx in w_idx_list: for w_idx in w_idx_list:
if state.interrupted: if state.interrupted or state.skipped:
break break
in_patch = img[..., h_idx: h_idx + tile, w_idx: w_idx + tile] in_patch = img[..., h_idx: h_idx + tile, w_idx: w_idx + tile]