* Fallback in case we don't have get_noise() (using the base class for some reason...)

2026-02-05 11:04:58 -05:00 · 2022-09-12 16:52:22 -04:00
parent 4a5a228fd8
commit 2a292d5b82
1 changed files with 4 additions and 1 deletions
--- 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])