Add files via upload

invoke-ai · Sep 4, 2022 · 02a6018 · 02a6018
1 parent 4406fd1
commit 02a6018
Showing 1 changed file with 30 additions and 0 deletions.
diff --git a/ldm/dream/server.py b/ldm/dream/server.py
@@ -80,13 +80,39 @@ def do_POST(self):
         seamless = 'seamless' in post_data
         cfgscale = float(post_data['cfgscale'])
         sampler_name  = post_data['sampler']
+        variation_amount = float(post_data['variation_amount'])
+        with_variations = post_data['with_variations']
         gfpgan_strength = float(post_data['gfpgan_strength']) if gfpgan_model_exists else 0
         upscale_level    = post_data['upscale_level']
         upscale_strength = post_data['upscale_strength']
         upscale = [int(upscale_level),float(upscale_strength)] if upscale_level != '' else None
         progress_images = 'progress_images' in post_data
         seed = self.model.seed if int(post_data['seed']) == -1 else int(post_data['seed'])
 
+        if with_variations != '':
+            parts = []
+            broken = False
+            for part in with_variations.split(','):
+                seed_and_weight = part.split(':')
+                if len(seed_and_weight) != 2:
+                    print(f'could not parse with_variation part "{part}"')
+                    broken = True
+                    break
+                try:
+                    vseed = int(seed_and_weight[0])
+                    vweight = float(seed_and_weight[1])
+                except ValueError:
+                    print(f'could not parse with_variation part "{part}"')
+                    broken = True
+                    break
+                parts.append([vseed, vweight])
+            if broken:
+                raise CanceledException
+            if len(parts) > 0:
+                with_variations = parts
+            else:
+                with_variations = None
+
         self.canceled.clear()
         print(f">> Request to generate with prompt: {prompt}")
         # In order to handle upscaled images, the PngWriter needs to maintain state
@@ -168,6 +194,8 @@ def image_progress(sample, step):
                                         height = height,
                                         seed = seed,
                                         steps = steps,
+                                        variation_amount = variation_amount,
+                                        with_variations = with_variations,
                                         gfpgan_strength = gfpgan_strength,
                                         upscale         = upscale,
                                         sampler_name    = sampler_name,
@@ -189,6 +217,8 @@ def image_progress(sample, step):
                                             cfg_scale  = cfgscale,
                                             seed       = seed,
                                             steps      = steps,
+                                            variation_amount = variation_amount,
+                                            with_variations = with_variations,
                                             sampler_name    = sampler_name,
                                             width      = width,
                                             height     = height,