Bug Fix Patch

ldm/dream/args.py

@@ -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(

ldm/dream/restoration/__init__.py

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

ldm/dream/restoration/base.py

@@ -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')