* Bug fixes

.gitignore

@@ -1,7 +1,7 @@
 # ignore default image save location and model symbolic link
 outputs/
 models/ldm/stable-diffusion-v1/model.ckpt
-ldm/restoration/codeformer/weights
+ldm/dream/restoration/codeformer/weights
 
 # ignore the Anaconda/Miniconda installer used while building Docker image
 anaconda.sh

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

ldm/dream/server.py

@@ -163,7 +163,7 @@ class DreamServer(BaseHTTPRequestHandler):
         # is complete. The upscaling replaces the original file, so the second
         # entry should not be inserted into the image list.
         # LS: This repeats code in dream.py
-        def image_done(image, seed, upscaled=False):
+        def image_done(image, seed, upscaled=False, first_seed=None):
             name = f'{prefix}.{seed}.png'
             iter_opt  = copy.copy(opt)
             if opt.variation_amount > 0:

scripts/dream.py

@@ -47,14 +47,14 @@ def main():
     # Loading Face Restoration and ESRGAN Modules
     try:
         gfpgan, codeformer, esrgan = None, None, None
-        from ldm.dream.restoration import Restoration
+        from ldm.dream.restoration.base import Restoration
         restoration = Restoration(opt.gfpgan_dir, opt.gfpgan_model_path, opt.esrgan_bg_tile)
         if opt.restore:
             gfpgan, codeformer = restoration.load_face_restore_models()
         else:
             print('>> Face restoration disabled')
         if opt.esrgan:
-            esrgan = restoration.load_ersgan()
+            esrgan = restoration.load_esrgan()
         else:
             print('>> Upscaling disabled')
     except (ModuleNotFoundError, ImportError):