Skip to content
Projects
Groups
Snippets
Help
Loading...
Help
Contribute to GitLab
Sign in / Register
Toggle navigation
S
stable-diffusion-webui
Project
Project
Details
Activity
Cycle Analytics
Repository
Repository
Files
Commits
Branches
Tags
Contributors
Graph
Compare
Charts
Issues
0
Issues
0
List
Board
Labels
Milestones
Merge Requests
0
Merge Requests
0
CI / CD
CI / CD
Pipelines
Jobs
Schedules
Charts
Wiki
Wiki
Snippets
Snippets
Members
Members
Collapse sidebar
Close sidebar
Activity
Graph
Charts
Create a new issue
Jobs
Commits
Issue Boards
Open sidebar
Administrator
stable-diffusion-webui
Commits
848ef919
Unverified
Commit
848ef919
authored
Jan 24, 2023
by
AUTOMATIC1111
Committed by
GitHub
Jan 24, 2023
Browse files
Options
Browse Files
Download
Plain Diff
Merge pull request #7144 from catboxanon/swinir-interrupt
Make SwinIR upscaler interruptible and skippable
parents
393e09c1
3c47b050
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
7 additions
and
1 deletion
+7
-1
swinir_model.py
extensions-builtin/SwinIR/scripts/swinir_model.py
+7
-1
No files found.
extensions-builtin/SwinIR/scripts/swinir_model.py
View file @
848ef919
...
@@ -8,7 +8,7 @@ from basicsr.utils.download_util import load_file_from_url
...
@@ -8,7 +8,7 @@ from basicsr.utils.download_util import load_file_from_url
from
tqdm
import
tqdm
from
tqdm
import
tqdm
from
modules
import
modelloader
,
devices
,
script_callbacks
,
shared
from
modules
import
modelloader
,
devices
,
script_callbacks
,
shared
from
modules.shared
import
cmd_opts
,
opts
from
modules.shared
import
cmd_opts
,
opts
,
state
from
swinir_model_arch
import
SwinIR
as
net
from
swinir_model_arch
import
SwinIR
as
net
from
swinir_model_arch_v2
import
Swin2SR
as
net2
from
swinir_model_arch_v2
import
Swin2SR
as
net2
from
modules.upscaler
import
Upscaler
,
UpscalerData
from
modules.upscaler
import
Upscaler
,
UpscalerData
...
@@ -145,7 +145,13 @@ def inference(img, model, tile, tile_overlap, window_size, scale):
...
@@ -145,7 +145,13 @@ 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
or
state
.
skipped
:
break
for
w_idx
in
w_idx_list
:
for
w_idx
in
w_idx_list
:
if
state
.
interrupted
or
state
.
skipped
:
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
]
out_patch
=
model
(
in_patch
)
out_patch
=
model
(
in_patch
)
out_patch_mask
=
torch
.
ones_like
(
out_patch
)
out_patch_mask
=
torch
.
ones_like
(
out_patch
)
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment