github/InvokeAI
1
1
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI.git synced 2026-02-05 10:29:07 -05:00
Code Issues Packages Projects Releases Wiki Activity

Merge branch 'development' into development

Browse Source
This commit is contained in:
Peter Baylies
2022-09-21 11:05:20 -04:00
committed by GitHub
parent 93001f48f7 069f91f930
commit 9903ce60f0
5 changed files with 6 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
ldm/dream/args.py
View File

@@ -573,14 +573,14 @@ class Args(object):
'-G',
'--gfpgan_strength',
type=float,
help='The strength at which to apply the GFPGAN model to the result, in order to improve faces.',
help='The strength at which to apply the face restoration to the result.',
default=0.0,
)
postprocessing_group.add_argument(
'-cf',
'--codeformer_fidelity',
type=float,
help='Takes values between 0 and 1. 0 produces high quality but low accuracy. 1 produces high accuracy but low quality.',
help='Used along with CodeFormer. Takes values between 0 and 1. 0 produces high quality but low accuracy. 1 produces high accuracy but low quality.',
default=0.75
)
postprocessing_group.add_argument(

1
ldm/dream/restoration/__init__.py Normal file
View File

@@ -0,0 +1 @@
from .base import Restoration

2
ldm/dream/restoration/base.py
View File

@@ -27,7 +27,7 @@ class Restoration():
return CodeFormerRestoration()
# Upscale Models
def load_ersgan(self):
def load_esrgan(self):
from ldm.dream.restoration.realesrgan import ESRGAN
esrgan = ESRGAN(self.esrgan_bg_tile)
print('>> ESRGAN Initialized')