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