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

Merge branch 'webui-sampler-fix' into main

Browse Source
This commit is contained in:
Lincoln Stein 2022-08-29 19:25:49 -04:00
commit 24ce56b3db
2 changed files with 4 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,

1
static/dream_web/index.html
View File

@ -32,6 +32,7 @@
<option value="k_dpm_2">KDPM_2</option> <option value="k_dpm_2">KDPM_2</option>
<option value="k_dpm_2_a">KDPM_2A</option> <option value="k_dpm_2_a">KDPM_2A</option>
<option value="k_euler">KEULER</option> <option value="k_euler">KEULER</option>
<option value="k_euler_a">KEULER_A</option>
<option value="k_heun">KHEUN</option> <option value="k_heun">KHEUN</option>
</select> </select>
<br> <br>