diff --git a/ldm/dream/server.py b/ldm/dream/server.py index db33f6d6a0..4170547be0 100644 --- a/ldm/dream/server.py +++ b/ldm/dream/server.py @@ -46,6 +46,7 @@ class DreamServer(BaseHTTPRequestHandler): width = int(post_data['width']) height = int(post_data['height']) cfgscale = float(post_data['cfgscale']) + sampler_name = post_data['sampler'] gfpgan_strength = float(post_data['gfpgan_strength']) upscale_level = post_data['upscale_level'] upscale_strength = post_data['upscale_strength'] @@ -118,6 +119,7 @@ class DreamServer(BaseHTTPRequestHandler): steps = steps, gfpgan_strength = gfpgan_strength, upscale = upscale, + sampler_name = sampler_name, step_callback=image_progress, image_callback=image_done) else: @@ -133,6 +135,7 @@ class DreamServer(BaseHTTPRequestHandler): cfg_scale = cfgscale, seed = seed, steps = steps, + sampler_name = sampler_name, gfpgan_strength=gfpgan_strength, upscale = upscale, step_callback=image_progress, diff --git a/static/dream_web/index.html b/static/dream_web/index.html index 5c036e90c3..158e559314 100644 --- a/static/dream_web/index.html +++ b/static/dream_web/index.html @@ -32,6 +32,7 @@ +