diff --git a/ldm/dream/args.py b/ldm/dream/args.py
index eb1913d1cf..2cb42d82a7 100644
--- a/ldm/dream/args.py
+++ b/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(
diff --git a/ldm/dream/restoration/__init__.py b/ldm/dream/restoration/__init__.py
new file mode 100644
index 0000000000..1a1de70ea2
--- /dev/null
+++ b/ldm/dream/restoration/__init__.py
@@ -0,0 +1 @@
+from .base import Restoration
\ No newline at end of file