prevent crashes due to uninitialized free_gpu_mem

ldm/generate.py

@@ -174,7 +174,8 @@ class Generate:
             config                = None,
             gfpgan=None,
             codeformer=None,
-            esrgan=None
+            esrgan=None,
+            free_gpu_mem=False,
     ):
         models              = OmegaConf.load(conf)
         mconfig             = models[model]
@@ -201,6 +202,7 @@ class Generate:
         self.gfpgan = gfpgan
         self.codeformer = codeformer
         self.esrgan = esrgan
+        self.free_gpu_mem = free_gpu_mem
 
         # Note that in previous versions, there was an option to pass the
         # device to Generate(). However the device was then ignored, so

scripts/dream.py

@@ -75,7 +75,8 @@ def main():
             precision      = opt.precision,
             gfpgan=gfpgan,
             codeformer=codeformer,
-            esrgan=esrgan
+            esrgan=esrgan,
+            free_gpu_mem=opt.free_gpu_mem,
             )
     except (FileNotFoundError, IOError, KeyError) as e:
         print(f'{e}. Aborting.')
@@ -104,8 +105,6 @@ def main():
 
     # preload the model
     gen.load_model()
-    #set additional option
-    gen.free_gpu_mem = opt.free_gpu_mem
 
     # web server loops forever
     if opt.web or opt.gui: