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

Double check for null and 0, and add a comment to indicate intent

Browse Source
This commit is contained in:
Sean McLellan 2022-08-26 04:05:09 -04:00
parent eb9f0be91a
commit 46a223f229
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
ldm/simplet2i.py
View File

@ -305,9 +305,12 @@ class T2I:
iter_images = next(images_iterator)
for image in iter_images:
try:
if gfpgan_strength > 0:
image = self._run_gfpgan(
image, gfpgan_strength
# if gfpgan strength is not none or less than 0.0 then don't use GFPGAN
# if the user specified -G x, and the --gfpgan wasn't specified, then
# the net result is a message gets printed - nothing else happens
if gfpgan_strength is not None and gfpgan_strength > 0.0:
image = self.gfpgan.generate(
image, gfpgan_strength=gfpgan_strength
)
except Exception as e:
print(