diff --git a/README.md b/README.md index f3cf91877f..179be569b2 100644 --- a/README.md +++ b/README.md @@ -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 diff --git a/scripts/dream.py b/scripts/dream.py index 0e888bf99c..e2825f8142 100755 --- a/scripts/dream.py +++ b/scripts/dream.py @@ -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)