Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add files via upload

Browse Source
This commit is contained in:
Web Dev 9000 2022-09-04 14:35:07 -07:00 committed by GitHub
parent 4406fd138d
commit 02a6018992
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 30 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
ldm/dream/server.py
View File

@ -80,6 +80,8 @@ class DreamServer(BaseHTTPRequestHandler):
seamless = 'seamless' in post_data seamless = 'seamless' in post_data
cfgscale = float(post_data['cfgscale']) cfgscale = float(post_data['cfgscale'])
sampler_name = post_data['sampler'] 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 gfpgan_strength = float(post_data['gfpgan_strength']) if gfpgan_model_exists else 0
upscale_level = post_data['upscale_level'] upscale_level = post_data['upscale_level']
upscale_strength = post_data['upscale_strength'] upscale_strength = post_data['upscale_strength']
@ -87,6 +89,30 @@ class DreamServer(BaseHTTPRequestHandler):
progress_images = 'progress_images' in post_data progress_images = 'progress_images' in post_data
seed = self.model.seed if int(post_data['seed']) == -1 else int(post_data['seed']) 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() self.canceled.clear()
print(f">> Request to generate with prompt: {prompt}") print(f">> Request to generate with prompt: {prompt}")
# In order to handle upscaled images, the PngWriter needs to maintain state # In order to handle upscaled images, the PngWriter needs to maintain state
@ -168,6 +194,8 @@ class DreamServer(BaseHTTPRequestHandler):
height = height, height = height,
seed = seed, seed = seed,
steps = steps, steps = steps,
variation_amount = variation_amount,
with_variations = with_variations,
gfpgan_strength = gfpgan_strength, gfpgan_strength = gfpgan_strength,
upscale = upscale, upscale = upscale,
sampler_name = sampler_name, sampler_name = sampler_name,
@ -189,6 +217,8 @@ class DreamServer(BaseHTTPRequestHandler):
cfg_scale = cfgscale, cfg_scale = cfgscale,
seed = seed, seed = seed,
steps = steps, steps = steps,
variation_amount = variation_amount,
with_variations = with_variations,
sampler_name = sampler_name, sampler_name = sampler_name,
width = width, width = width,
height = height, height = height,