From 6fb2f1ed6ed08a65fde4fefa528136c3bcccd006 Mon Sep 17 00:00:00 2001 From: Lincoln Stein Date: Mon, 29 Aug 2022 14:06:18 -0400 Subject: [PATCH] fixes WebUI so that the selected sampler is actually applied --- ldm/dream/server.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/ldm/dream/server.py b/ldm/dream/server.py index efe3139f74..e41fa6c667 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,