Add files via upload

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,