Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

Remove the redundancy, better logging

Browse Source
This commit is contained in:
Sean McLellan 2022-08-25 23:48:35 -04:00
parent 3a30a8f2d2
commit cb86b9ae6e
2 changed files with 3 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
ldm/simplet2i.py
View File

@ -399,8 +399,8 @@ The vast majority of these arguments default to reasonable values.
try:
if gfpgan_strength > 0:
image = self._run_gfpgan(image, gfpgan_strength)
except Exception:
print(f"Error running GFPGAN - Your image was not enhanced.")
except Exception as e:
print(f"Error running GFPGAN - Your image was not enhanced.\n{e}")
images.append(image)
return images

4
scripts/dream.py
View File

@ -81,9 +81,7 @@ def main():
sys.path.append(os.path.abspath(opt.gfpgan_dir))
from gfpgan import GFPGANer
bg_upsampler = None
if opt.gfpgan_bg_upsampler == 'realesrgan':
bg_upsampler = load_gfpgan_bg_upsampler(opt.gfpgan_bg_upsampler, opt.gfpgan_bg_tile)
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)
except Exception: