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 @@
+