Fix not being able to disable bgupscaler; update readme

This commit is contained in:
Sean McLellan 2022-08-25 23:39:03 -04:00
parent 60ed004328
commit 3a30a8f2d2
2 changed files with 2 additions and 2 deletions

View File

@ -88,7 +88,7 @@ and optionally upscale the image to a higher resolution.
To use the ability, clone the [GFPGAN repository](https://github.com/TencentARC/GFPGAN) and follow their
installation instructions. By default, we expect GFPGAN to be installed in a 'GFPGAN' sibling directory.
You may also want to install Real-ESRGAN, if you want to enhance non-face regions in the image by installing
You may also want to install Real-ESRGAN, if you want to enhance non-face regions in the image, by installing
the pip Real-ESRGAN package.
```
pip install realesrgan

View File

@ -82,7 +82,7 @@ def main():
from gfpgan import GFPGANer
bg_upsampler = None
if opt.gfpgan_bg_upsampler is not None:
if opt.gfpgan_bg_upsampler == 'realesrgan':
bg_upsampler = load_gfpgan_bg_upsampler(opt.gfpgan_bg_upsampler, opt.gfpgan_bg_tile)
t2i.gfpgan = GFPGANer(model_path=model_path, upscale=opt.gfpgan_upscale, arch='clean', channel_multiplier=2, bg_upsampler=bg_upsampler)