From 2a292d5b823b3ed2c19d5c2c6cf9bc0afc7bdbeb Mon Sep 17 00:00:00 2001
From: Peter Baylies <pbaylies@gmail.com>
Date: Mon, 12 Sep 2022 16:52:22 -0400
Subject: [PATCH]  * Fallback in case we don't have get_noise() (using the base
 class for some reason...)

---
 ldm/dream/generator/base.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/ldm/dream/generator/base.py b/ldm/dream/generator/base.py
index 38d2e373af..92e5a62105 100644
--- a/ldm/dream/generator/base.py
+++ b/ldm/dream/generator/base.py
@@ -66,7 +66,10 @@ class Generator():
                     x_T = initial_noise
                 else:
                     seed_everything(seed)
-                    x_T = self.get_noise(width,height)
+                    try:
+                        x_T = self.get_noise(width,height)
+                    except:
+                        pass
 
                 image = make_image(x_T)
                 results.append([image, seed])