Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
ModelZoo
DiffBIR_pytorch
Commits
8aa292c6
Commit
8aa292c6
authored
Sep 19, 2023
by
ziyannchen
Browse files
fix RealESRGAN model downloading url
parent
fa2a0969
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
6 additions
and
1 deletion
+6
-1
utils/realesrgan/realesrganer.py
utils/realesrgan/realesrganer.py
+6
-1
No files found.
utils/realesrgan/realesrganer.py
View file @
8aa292c6
...
...
@@ -313,6 +313,11 @@ def set_realesrgan(bg_tile, device, scale=2):
if
not
True
in
[
gpu
in
torch
.
cuda
.
get_device_name
(
0
)
for
gpu
in
no_half_gpu_list
]:
use_half
=
True
model_url
=
{
2
:
'https://github.com/xinntao/Real-ESRGAN/releases/download/v0.2.1/RealESRGAN_x2plus.pth'
,
4
:
'https://github.com/xinntao/Real-ESRGAN/releases/download/v0.1.0/RealESRGAN_x4plus.pth'
}
model
=
RRDBNet
(
num_in_ch
=
3
,
num_out_ch
=
3
,
...
...
@@ -323,7 +328,7 @@ def set_realesrgan(bg_tile, device, scale=2):
)
upsampler
=
RealESRGANer
(
scale
=
scale
,
model_path
=
f
"https://github.com/xinntao/Real-ESRGAN/releases/download/v0.1.0/RealESRGAN_x
{
scale
}
plus.pth"
,
model_path
=
model_url
[
scale
]
,
model
=
model
,
tile
=
bg_tile
,
tile_pad
=
40
,
...
...
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