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

fixes WebUI so that the selected sampler is actually applied

Browse Source
This commit is contained in:
Lincoln Stein 2022-08-29 14:06:18 -04:00
parent 38701a6d7b
commit 6fb2f1ed6e
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
ldm/dream/server.py
View File

@ -46,6 +46,7 @@ class DreamServer(BaseHTTPRequestHandler):
width = int(post_data['width']) width = int(post_data['width'])
height = int(post_data['height']) height = int(post_data['height'])
cfgscale = float(post_data['cfgscale']) cfgscale = float(post_data['cfgscale'])
sampler_name = post_data['sampler']
gfpgan_strength = float(post_data['gfpgan_strength']) gfpgan_strength = float(post_data['gfpgan_strength'])
upscale_level = post_data['upscale_level'] upscale_level = post_data['upscale_level']
upscale_strength = post_data['upscale_strength'] upscale_strength = post_data['upscale_strength']
@ -118,6 +119,7 @@ class DreamServer(BaseHTTPRequestHandler):
steps = steps, steps = steps,
gfpgan_strength = gfpgan_strength, gfpgan_strength = gfpgan_strength,
upscale = upscale, upscale = upscale,
sampler_name = sampler_name,
step_callback=image_progress, step_callback=image_progress,
image_callback=image_done) image_callback=image_done)
else: else:
@ -133,6 +135,7 @@ class DreamServer(BaseHTTPRequestHandler):
cfg_scale = cfgscale, cfg_scale = cfgscale,
seed = seed, seed = seed,
steps = steps, steps = steps,
sampler_name = sampler_name,
gfpgan_strength=gfpgan_strength, gfpgan_strength=gfpgan_strength,
upscale = upscale, upscale = upscale,
step_callback=image_progress, step_callback=image_progress,